【BZOJ】3430 [Usaco2014 Jan]Ski Course Rating 并查集

题目传送门

这题嘛，我其实在普及组就做过了……好吧当时的题目没有给出多个起点，直接用二分+BFS的暴力就可以过去了。

但是这题有多个起点，还让我们求的是所有起点的难度和（题目意思真难懂……），那我们就不能这么瞎搞了。

考虑每个节点向它下方和右方建边，然后将所有的边sort一遍，对每条边的两端做并查集的合并。

每次选取一条边，如果两端的联通块大小大于题目给出的t且其中一端的联通块大小小于t，就更新一下答案。

因为我们对于边权排过序了，所以每次更新答案时最大的差值就是当前的边权。

然后这题就AC啦（突然发现这是一道水题……）。

附上AC代码：

 

#include <cstdio>
#include <cctype>
#include <algorithm>
using namespace std;

const int N=510,dx[2]={0,1},dy[2]={1,0};
struct side{
	int x,y,w;
	bool operator < (const side lyf) const {return w<lyf.w;}
}s[N*N*2];
int n,m,t,map[N][N],wz[N][N],num,f[N*N],sz[N*N],o,c[N*N],g;
long long ans;

inline char nc(){
	static char ch[100010],*p1=ch,*p2=ch;
	return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
	static char c=nc();int f=1;
	for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
	for (a=0;isdigit(c);a=a*10+c-'0',c=nc());
	a*=f;return;
}

inline int gf(int x){return x==f[x]?x:f[x]=gf(f[x]);}

int main(void){
	read(n),read(m),read(t);
	for (int i=1; i<=n; ++i)
		for (int j=1; j<=m; ++j)
			read(map[i][j]),wz[i][j]=++g,f[g]=g,sz[g]=1;
	for (int i=1; i<=n; ++i)
		for (int j=1; j<=m; ++j){
			read(o);
			if (o) c[wz[i][j]]=1;
			for (int k=0; k<=1; ++k){
				int fx=i+dx[k],fy=j+dy[k];
				if (fx<n+1&&fy<m+1) s[++num]=(side){wz[i][j],wz[fx][fy],abs(map[i][j]-map[fx][fy])};
			}
		}
	sort(s+1,s+1+num);
	for (int i=1; i<=num; ++i){
		int fx=gf(s[i].x),fy=gf(s[i].y);
		if (fx==fy) continue;
		if (sz[fx]+sz[fy]>=t){
			if (sz[fx]<t) ans+=1ll*c[fx]*s[i].w;
			if (sz[fy]<t) ans+=1ll*c[fy]*s[i].w;
		}
		if (sz[fx]<sz[fy]) swap(fx,fy);
		f[fy]=fx,sz[fx]+=sz[fy],c[fx]+=c[fy];
	}
	printf("%lld",ans);
	return 0;
}