2019.2.14 t3 车辆销售

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

 #include <cstdio>
 #include <iostream>
 #include <cstring>
 #include <algorithm>
 #include <cmath>
 #include <cctype>
 using namespace std;

 #define LL long long
 #define res register int
 inline int read()
 {
     int x(),f(); char ch;
     while(!isdigit(ch=getchar())) if(ch=='-') f=-;
     while(isdigit(ch)) x=x*+ch-'',ch=getchar();
     return f*x;
 }
 int n,m;
 const int N=+;
 int head[N],nxt[N<<],ver[N<<],tot;
 struct E{
     int u,v,w;
 }e[N];
 int vis[N],fa[N];
 LL w[N],size[N],val[N];
 inline bool cmp(E a,E b) {return a.w>b.w;}
 inline void add(int x,int y) {
     ver[++tot]=y; nxt[tot]=head[x]; head[x]=tot;
     ver[++tot]=x; nxt[tot]=head[y]; head[y]=tot;
 }    

 inline int get(int x) {
     if(x==fa[x]) return x; return fa[x]=get(fa[x]);
 }

 int cnt();
 inline void Kruskal()
 {
     cnt=n;
     sort(e+,e+m+,cmp);
     for(res i= ; i<=n ; i++) fa[i]=i;
     for(res i= ; i<=m ; i++)
     {
         int x=get(e[i].u),y=get(e[i].v),z=e[i].w;
         if(x==y) continue;
         val[++cnt]=z;
         fa[cnt]=fa[x]=fa[y]=cnt;
         add(x,cnt); add(y,cnt);
         if(cnt==*n-) break;
     }
 }

 inline void dfs(int x,int f)
 {
     vis[x]=;
     for(res i(head[x]) ; i ; i=nxt[i])
     {
         int y=ver[i]; if(f==y) continue;
         dfs(y,x); size[x]+=size[y];
     }
 }

 inline void dp(int x,int f)
 {
     for(res i(head[x]) ; i ; i=nxt[i])
     {
         int y=ver[i]; if(f==y) continue;
         if(val[x]==val[y]) size[y]=size[x];
         w[y]=w[x]+(size[x]-size[y])*(size[x]-size[y]);
         dp(y,x);
     }
 }

 inline void work()
 {
     for(res i= ; i<=n ; i++) size[i]=;
     for(res i= ; i<=cnt ; i++)
     {
         if(vis[i]) continue;
         int x=get(i);
         dfs(x,);
         dp(x,);
     }
 }

 int main()
 {
     freopen("car.in","r",stdin);
     freopen("car.out","w",stdout);
     n=read(); m=read()-;
     for(res i= ; i<=m ; i++) {
         e[i].u=read(); e[i].v=read(); e[i].w=read();
     }
     Kruskal();
     work();
     for(res i= ; i<=n ; i++) cout<<w[i]<<" ";//printf("%d ",w[i]);
     puts("");
     return ;
 }