ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

CF1120D Power Tree

2022-01-02 22:01:31  阅读:173  来源: 互联网

标签:Bg struct Power int Tree long Edge CF1120D define


题面传送门
首先不难想到把这个叶子的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));
}

标签:Bg,struct,Power,int,Tree,long,Edge,CF1120D,define
来源: https://www.cnblogs.com/275307894a/p/15758460.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有