题意:
现在有n个点,m条无向边,然后有q个询问。
给出a,b,d,代表从a到b需要花费d的时间,从而b到a也是一眼。
然后对于q个询问,给出x,代表这个人能够最大容忍的花费在车上的时间为x。
思路:
花了好久,终于想出做法了,但是后来又T了。。(原因有一部分没经过思考,写的太暴力了。。)
首先,保存每一条边的起点和终点,还有边长,首先按照边的权值来排序。然后对于q个询问,我们事先把询问都存下来,然后进行排序。
因为这里的方案数就是每个连通分量中点的个数*(点的个数-1),所以我们的目的就是知道点的个数就好了。
首先我想到的第一个问题是怎么把点全都合并起来呢?怎么判断两个点之间是否有边相连呢? Answer:用并查集。
只需要把起点和终点不停地合并起来,最后再进行一次for,判断到底哪个点是祖先节点,然后再加上祖先节点下面的所包含的所有的节点数量。
但是这个找节点数的做法会超时的。假设我们这里不算上(A,B) (B,A)是不同情况,那么我们每次只需加上新增加某些点会多出的方案数就好了,最后再乘以2就是当前答案。
*要注意对询问进行排序后,注意保存它的序号,因为最后输出的时候还得按照题目中的那个要求来的。所以记得保存序号!!
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
#define maxn 20010
typedef __int64 ll;
int par[maxn],num[maxn];
int ans[5010];
int cnt=0;
struct node{
int a;
int b;
int d;
}t[100010];
struct nn{
int idx;
int x;
}que[5010];
bool cmp(node a,node b){
return a.d<b.d;
}
bool cmp2(nn a,nn b){
return a.x<b.x;
}
void init(){
for(int i=0;i<20010;i++) par[i]=i,num[i]=1;
}
int find(int x){
if(par[x]!=x) par[x]=find(par[x]);
return par[x];
}
void merge(int x,int y){
int tx=find(x);
int ty=find(y);
if(tx!=ty){
cnt+=num[tx]*num[ty]; //使用了每次不断累加的方法,更加优化!
par[tx]=ty;
num[ty]+=num[tx];
}
}
int main(){
int T;
scanf("%d",&T);
while(T--){
init();
memset(que,0,sizeof(que));
int n,m,Q;
scanf("%d%d%d",&n,&m,&Q);
for(int i=0;i<m;i++){
int a,b,d;
scanf("%d%d%d",&a,&b,&d);
t[i+1].a=a; t[i+1].b=b; t[i+1].d=d;
}
sort(t+1,t+1+m,cmp);
for(int i=0;i<Q;i++){
int x;
scanf("%d",&x);
que[i].x=x;
que[i].idx=i;
}
sort(que,que+Q,cmp2);
int tmp=1;
cnt=0;
memset(ans,0,sizeof(ans));
int tt=0;
for(int i=0;i<Q;i++){
int x=que[i].x;
int idd=que[i].idx;
for(int j=tmp;j<=m;j++){
if(t[j].d<=x){
int s=t[j].a,e=t[j].b;
merge(s,e);
}
else{
tmp=j;
break;
}
}
ans[idd]=cnt*2;
}
for(int i=0;i<Q;i++) printf("%d\n",ans[i]);
}
}
/*
1
5 5 3
2 3 6334
1 5 15724
3 5 5705
4 3 12382
1 3 21726
13000
10000
6000
*/