标签：head ch 14 int t3 2019.2 tot inline include

• 用算法求最大生成树，在并查集合并时，把原本的一个根连向另一个 根改成两个根都连向一个新建的节点，并把当前正在处理的边的权值赋给这个新 节点做点权。这样形成的结构会是一棵树。 一个点的答案大致上是树的根到自己的路径上，相邻两个节点的子树叶节点 数的平方和。需要注意的是父子两个节点权值相同的情况，这个部分需要特殊处理。

 1 #include <cstdio>
 2 #include <iostream>
 3 #include <cstring>
 4 #include <algorithm>
 5 #include <cmath>
 6 #include <cctype>
 7 using namespace std;
 8 
 9 #define LL long long
10 #define res register int
11 inline int read()
12 {
13     int x(0),f(1); char ch;
14     while(!isdigit(ch=getchar())) if(ch=='-') f=-1;
15     while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
16     return f*x;
17 }
18 int n,m;
19 const int N=500000+5;
20 int head[N],nxt[N<<1],ver[N<<1],tot;
21 struct E{
22     int u,v,w;
23 }e[N];
24 int vis[N],fa[N];
25 LL w[N],size[N],val[N];
26 inline bool cmp(E a,E b) {return a.w>b.w;}
27 inline void add(int x,int y) {
28     ver[++tot]=y; nxt[tot]=head[x]; head[x]=tot;
29     ver[++tot]=x; nxt[tot]=head[y]; head[y]=tot;
30 }    
31 
32 inline int get(int x) {
33     if(x==fa[x]) return x; return fa[x]=get(fa[x]);
34 }
35 
36 int cnt(0);
37 inline void Kruskal()
38 {
39     cnt=n;
40     sort(e+1,e+m+1,cmp);
41     for(res i=1 ; i<=n ; i++) fa[i]=i;
42     for(res i=1 ; i<=m ; i++)
43     {
44         int x=get(e[i].u),y=get(e[i].v),z=e[i].w;
45         if(x==y) continue;
46         val[++cnt]=z;
47         fa[cnt]=fa[x]=fa[y]=cnt;
48         add(x,cnt); add(y,cnt);
49         if(cnt==2*n-1) break;
50     }
51 }
52 
53 inline void dfs(int x,int f)
54 {
55     vis[x]=1;
56     for(res i(head[x]) ; i ; i=nxt[i])
57     {
58         int y=ver[i]; if(f==y) continue;
59         dfs(y,x); size[x]+=size[y];
60     }
61 }
62 
63 inline void dp(int x,int f)
64 {
65     for(res i(head[x]) ; i ; i=nxt[i])
66     {
67         int y=ver[i]; if(f==y) continue;
68         if(val[x]==val[y]) size[y]=size[x];
69         w[y]=w[x]+(size[x]-size[y])*(size[x]-size[y]); 
70         dp(y,x);
71     }
72 }
73 
74 inline void work()
75 {
76     for(res i=1 ; i<=n ; i++) size[i]=1;
77     for(res i=1 ; i<=cnt ; i++)
78     {
79         if(vis[i]) continue;
80         int x=get(i);
81         dfs(x,0); 
82         dp(x,0);
83     }
84 }
85 
86 int main()
87 {
88     freopen("car.in","r",stdin);
89     freopen("car.out","w",stdout);
90     n=read(); m=read()-1;
91     for(res i=1 ; i<=m ; i++) {
92         e[i].u=read(); e[i].v=read(); e[i].w=read();
93     }
94     Kruskal();
95     work();
96     for(res i=1 ; i<=n ; i++) cout<<w[i]<<" ";//printf("%d ",w[i]);
97     puts("");
98     return 0;
99 }

标签：head,ch,14,int,t3,2019.2,tot,inline,include
来源： https://www.cnblogs.com/wmq12138/p/10380846.html

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