解题思路:首先我们很容易就能想到用一个并查集来维护一个集合的连通性,也就是说第一个操作,所以难点在于第二个操作,如何通过一个点去将该点所在连通块中的所有的点的权值都加上一个数t,这里就是要给每个点设置一个权值,用当前点到根节点的距离来表示当前点接受或发送信息的次数,注意这里到根节点的距离不能加上根节点的值,因为会对后面的操作造成不便,那现在问题来了,如何在合并的时候进行值的更新来?也就是我们熟悉的find操作,这个具体可以看代码,变化不是很大,一看就能明白,然后是比较关键的合并2个集合的操作,这个操作可以有2种方法.
1、假设现在有2棵树,根节点分别是fu,fv,要将fu合并到fv上,如果直接就合并,那fu子树上的每个点的权值都要加上点fv的权值,这显然是不对的,多以我们可以事先将点fu的权值减去点fv的权值,这里就是我们在算到根节点的距离为什么不能加上根节点的权值的原因了,如果用根节点的值对它的子节点的值进行更新了,那在进行2棵树合并的时候,就得把每个子节点的权值都减去fv点的权值,这就变的很复杂了。
2、合并子树的时候,我们也可以新建一个结点,让这个结点成为当前要合并的2个结点的父亲结点,这样话空间就需要开的大一点才行了。
上代码:
第一种方法:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <set>
using namespace std;
const int N=1e4+10;
int p[N],d[N];
int n,m;
int find(int x)
{
if(x==p[x]||p[p[x]]==p[x])
return p[x];
int r=find(p[x]);
d[x]+=d[p[x]];
p[x]=r;
return r;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
p[i]=i;
for(int i=1;i<=m;i++)
{
int op;
cin>>op;
if(op==1)
{
int u,v;
cin>>u>>v;
int fu=find(u),fv=find(v);
if(fu!=fv)
{
d[fu]-=d[fv];
p[fu]=fv;
}
}
else
{
int u,t;
cin>>u>>t;
int fu=find(u);
d[fu]+=t;
}
}
for(int i=1;i<=n;i++)
if(find(i)==i)
cout<<d[i]<<" ";
else
cout<<d[i]+d[find(i)]<<" ";
cout<<endl;
return 0;
}
第2种方法:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <set>
using namespace std;
const int N=2e4+10;
int p[N],d[N];
int n,m;
int find(int x)
{
if(x==p[x]||p[p[x]]==p[x])
return p[x];
int r=find(p[x]);
d[x]+=d[p[x]];
p[x]=r;
return r;
}
int main()
{
cin>>n>>m;
for(int i=1;i<N;i++)
p[i]=i;
int k=n;
for(int i=1;i<=m;i++)
{
int op;
cin>>op;
if(op==1)
{
int u,v;
cin>>u>>v;
int fu=find(u),fv=find(v);
if(fu!=fv)
{
k++;
p[fu]=p[fv]=k;
// d[fu]-=d[fv];
// p[fu]=fv;
}
}
else
{
int u,t;
cin>>u>>t;
int fu=find(u);
d[fu]+=t;
}
}
for(int i=1;i<=n;i++)
if(find(i)==i)
cout<<d[i]<<" ";
else
cout<<d[i]+d[find(i)]<<" ";
cout<<endl;
return 0;
}