CF K. City

https://codeforces.com/gym/103145/problem/K

 

一开始看到有环，就想着缩点了，但是对于每次询问，如果都建一个新图，必定会TLE

所以采用离线处理的方法，把询问按照由大到小来排序，这样就可以边询问边建图。

可以采用并查集的方法，开个记录每个集合中元素的个数。当两个集合合并时，产生的新数对个数就是两个集合大小之积。

合并的次序很重要，一定要先处理完再去合并：

        //这个顺序重要,一定要是先处理完再合并，不要以为fa没影响。
        //因为先合并的话，如果某个元素的祖先刚好是自己，那么这个元素就是先跑到了别人家再合并的，大小也是用的别人的大小

最后还要注意加速cin，因为输入的数据量比较大，直接cin会超时QWQ，或者scanf也可以

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define int ll 
const int maxn=3e5;
int cnt,fa[maxn];
ll ans[maxn],setnum[maxn];

struct node{
	int fr,to,w;
}e[maxn];

struct query{
	int id,d;
}p[maxn];

void add(int u,int v,int w){
	e[++cnt].fr=u;
	e[cnt].to=v;
	e[cnt].w=w;
}

bool cmp_e(node a,node b){
	return a.w>b.w;
}

bool cmp_q(query a,query b){
	return a.d>b.d;
}

int getfa(int x){
	return fa[x]==x?x:fa[x]=getfa(fa[x]);
}

signed main(){
	ios::sync_with_stdio(false);	//是个指令，所以不能放在头文件里 
	cin.tie(0);
	cout.tie(0);
	int t;
	cin>>t;
	while(t--)
	{
		int n,m,q;
		cin>>n>>m>>q;
		cnt=0;
		for(int i=1;i<=n;i++) fa[i]=i;
		for(int i=1;i<=m;i++)
		{
			int u,v,w;
			cin>>u>>v>>w;
			add(u,v,w);
		}
		for(int i=1;i<=q;i++)
		{
			cin>>p[i].d;
			p[i].id=i;
		}
		sort(e+1,e+1+cnt,cmp_e);
		sort(p+1,p+1+q,cmp_q);
		int k=1;
		ll sum=0;
		for(int i=1;i<=n;i++) setnum[i]=1;
		for(int i=1;i<=q;i++)
		{
			while(e[k].w>=p[i].d&&k<=cnt)
			{
				int u=e[k].fr,v=e[k].to;
				k++;
				if(getfa(u)!=getfa(v))
				{
//					cout<<"!!"<<getfa(u)<<" "<<fa[u]<<endl;
					sum+=setnum[fa[v]]*setnum[fa[u]];	
					setnum[fa[v]]+=setnum[fa[u]]; 
					fa[fa[u]]=fa[v];
		//这个顺序重要,一定要是先处理完再合并，不要以为fa没影响。
		//因为先合并的话，如果某个元素的祖先刚好是自己，那么这个元素就是先跑到了别人家再合并的，大小也是用的别人的大小 
				}
			}
			ans[p[i].id]=sum;
		}
		for(int i=1;i<=q;i++) cout<<ans[i]<<endl;
	}
	return 0; 
}