ACM : Travel-并查集-最小生成树 + 离线-解题报告

#include"iostream"
#include"cstdio"
#include"cstring"
#include"cmath"
#include"algorithm"
using namespace std;
const int MX=2222222;
long long ans[MX];
int P[MX], num[MX];

struct node {
	int u, v, cost;
} side[MX];

bool cmp_side(node a,node b) {
	return a.cost < b.cost;
}

struct Que {
	int id, x; //id 记录输入时候的顺序 x记录查询值 
} Q[MX];

bool cmp_que(Que a,Que b) {
	return a.x<b.x;
}

int find(int x) {
	return P[x]==x?x:(P[x]=find(P[x]));
}

int main() {
	int T;
	scanf("%d", &T);
	while(T--) {
		int n, m, q;
		scanf("%d%d%d", &n, &m, &q);
		for(int i=1; i<=n; i++) {
			P[i]=i;
			num[i]=1;       //初始化连通分量
		}
		for(int i=1; i<=m; i++) {	//把无向边加入队列
			scanf("%d%d%d", &side[i].u, &side[i].v, &side[i].cost);
		}
		sort(side+1,side+m+1,cmp_side);//升序排序无向边
		for(int i=1; i<=q; i++) {	//添加查询队列
			Q[i].id=i;
			scanf("%d",&Q[i].x);
		}
		sort(Q+1,Q+q+1,cmp_que);		//按照查询复杂度排序答案。
		int k=1;
		long long s=0;
		for(int i=1; i<=m; i++) {
			while(k<=q&&side[i].cost>Q[k].x)ans[Q[k++].id]=s; //把答案按照ID的顺序保存到ans数组中
			int rt1=find(side[i].u),rt2=find(side[i].v);
			if(rt1!=rt2) {
				long long n1=num[rt1],n2=num[rt2];
				//每个拥有n个节点的联通块有 n*(n-1)个联通分量 （对点）
				s=s-n1*(n1-1)-n2 *(n2-1)+(n1+n2)*(n1+n2-1);//去掉两个根的联通量在加上总和的联通分量
				P[rt1]=rt2;
				num[rt2]+=num[rt1];//加上节点的数量
			}
		}
		//ans[Q[k].id]=s; // 【这里居然看在上面读入ans之后还没有到最后一个，后面再测试数据才发现了这个问题。。。】
		while(k<=q)  
			ans[Q[k++].id]=s;
		for(int i=1; i<=q; i++)
			printf("%I64d\n",ans[i]); //按照离线保存的实际查询顺序输出
	}
	return 0;
}