题面传送门
首先不难想到把这个叶子的dfs序推成区间。然后对应了一个区间加
区间加容易让人想到差分,变成单点加。
又因为差分后全部的和为0,也就是说如果把这个看成一条边,那么联通块是可以全部调整成0的。
所以做一个MST即可,注意判断可以在其中一中里面的边。
code:
#include<bits/stdc++.h>
#define I inline
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define abs(x) ((x)>0?(x):-(x))
#define re register
#define RI re int
#define ll long long
#define db double
#define lb long db
#define N 200000
#define M 100000
#define mod 1000000007
#define Mod (mod-1)
#define eps (1e-5)
#define U unsigned int
#define it iterator
#define Gc() getchar()
#define Me(x,y) memset(x,y,sizeof(x))
#define d(x,y) (m*x+(y))
#define R(n) (rand()*rand()%(n)+1)
#define Pc(x) putchar(x)
using namespace std;
int n,m,W[N+5],Bg[N+5],En[N+5],cnt,fa[N+5],x,y,T[N+5],Fl[N+5];ll ToT;
struct yyy{int to,z;};struct ljb{int head,h[N+5];yyy f[N+5<<1];I void add(int x,int y){f[++head]=(yyy){y,h[x]};h[x]=head;}}s;
I void dfs(int x,int La){yyy tmp;Bg[x]=1e9;En[x]=0;for(RI i=s.h[x];i;i=tmp.z) tmp=s.f[i],tmp.to^La&&(dfs(tmp.to,x),Bg[x]=min(Bg[x],Bg[tmp.to]),En[x]=max(En[x],En[tmp.to]));Bg[x]>n&&(Bg[x]=En[x]=++cnt);}
struct Edge{int x,y,w,id;}S[N+5];I bool cmp(Edge x,Edge y){return x.w<y.w;}I int GF(int x){return fa[x]^x?fa[x]=GF(fa[x]):x;}
int main(){
freopen("1.in","r",stdin);
RI i,j,h;scanf("%d",&n);for(i=1;i<=n;i++) scanf("%d",&W[i]),fa[i]=i;for(i=1;i<n;i++)scanf("%d%d",&x,&y),s.add(x,y),s.add(y,x);dfs(1,0);
for(i=1;i<=n;i++) S[i]=(Edge){Bg[i],En[i]+1,W[i],i};sort(S+1,S+n+1,cmp);for(i=1;i<=n;i=j){
for(j=i+1;j<=n;j++) if(S[j].w^S[i].w) break;for(h=i;h<j;h++) GF(S[h].x)^GF(S[h].y)&&(Fl[S[h].id]=1);for(h=i;h<j;h++) GF(S[h].x)^GF(S[h].y)&&(fa[GF(S[h].x)]=GF(S[h].y),ToT+=S[h].w);
}printf("%lld ",ToT);for(cnt=0,i=1;i<=n;i++) cnt+=Fl[i];printf("%d\n",cnt);for(i=1;i<=n;i++) Fl[i]&&(printf("%d ",i));
}