hdu 3038（How Many Answers Are Wrong）+3047（Zjnu Stadium）（种类并查集）

hdu 3038：http://acm.hdu.edu.cn/showproblem.php?pid=3038

hdu 3047：http://acm.hdu.edu.cn/showproblem.php?pid=3047

如果之前没见过种类并查集的话，估计第一反应都是线段树吧...

这两道题都是在压缩路径的时候，利用递归，求得节点到根的距离。

用sum[i]表示节点i到根节点的距离,f[i]记录i的根节点。

f[rb]=ra;

sum[rb]=sum[a]-sum[b]+c;()



hdu 3038 代码（去掉注释那行可以秒过3047）：

#include <limits.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <algorithm>
#include <iostream>
#include <iterator>
#include <sstream>
#include <queue>
#include <stack>
#include <string>
#include <vector>
#include <list>
#include <set>
//#define ONLINE_JUDGE
#define eps 1e-6
#define INF 0x7fffffff                                          //INT_MAX
#define inf 0x3f3f3f3f                                          //int??????????????????
#define FOR(i,a) for((i)=0;i<(a);(i)++)                          //[i,a);
#define MEM(a) (memset((a),0,sizeof(a)))
#define sfs(a) scanf("%s",a)
#define sf(a) scanf("%d",&a)
#define sfI(a) scanf("%I64d",&a)
#define pf(a) printf("%d\n",a)
#define pfI(a) printf("%I64d\n",a)
#define pfs(a) printf("%s\n",a)
#define sfd(a,b) scanf("%d%d",&a,&b)
#define sft(a,b,c)scanf("%d%d%d",&a,&b,&c)
#define for1(i,a,b) for(int i=(a);i<b;i++)
#define for2(i,a,b) for(int i=(a);i<=b;i++)
#define for3(i,a,b)for(int i=(b);i>=a;i--)
#define MEM1(a) memset(a,0,sizeof(a))
#define MEM2(a) memset(a,-1,sizeof(a))
#define MEM3(a) memset(a,0x3f,sizeof(a))
#define MEMS(a) memset(a,'\0',sizeof(a))
#define LL __int64
const double PI = acos(-1.0);
template<class T> T gcd(T a, T b) { return b ? gcd(b, a % b) : a; }
template<class T> T lcm(T a, T b) { return a / gcd(a, b) * b; }
template<class T> inline T Min(T a, T b) { return a < b ? a : b; }
template<class T> inline T Max(T a, T b) { return a > b ? a : b; }
using namespace std;
template<class T>
T Mint(T a, T b, T c) {
if (a>b) {
if (c>b)
return b;
return c;
}
if (c>a)
return a;
return c;
}
template<class T>
T Maxt(T a, T b, T c) {
if (a>b) {
if (c>a)
return c;
return a;
}
else if (c > b)
return c;
return b;
}

const int maxn=200010;
int T,n,m;
int ans,f[maxn],sum[maxn];

void Make_Set(int x){
f[x]=x;
sum[x]=0;
}

int find(int x){
if(x!=f[x]){
int tmp=f[x];
f[x]=find(f[x]);
sum[x]+=sum[tmp];
}
return f[x];
}

void Union(int a,int b,int c){
int ra=find(a);
int rb=find(b);
if(ra==rb){
if(sum[b]-sum[a]!=c)
ans++;
}
else{
f[rb]=ra;
sum[rb]=sum[a]-sum[b]+c;
}
}

int main(){
#ifndef ONLINE_JUDGE
freopen("test.in","r",stdin);
freopen("test.out","w",stdout);
#endif
while(~sfd(n,m)){
for2(i,0,n)
Make_Set(i);
int a,b,c;
ans=0;
while(m--){
sft(a,b,c);
a--;//去掉这一行可以秒掉hdu 3047
Union(a,b,c);
}
pf(ans);
}
return 0;
}