Uva1665岛屿——并查集

题意：

输入一个n*m矩阵，每个格子都有一个正整数，再输入T个整数ti，对于每个ti，输出大于ti的正整数组成多少个四连块。

思路：

首先想到的就是枚举处理每个ti，对每个ti而言，问题都是一个求多少个连通块，可以用并查集。

#include <cstdio>
#include <queue>
#include <vector>
#include <cstring>
#include <algorithm>
#define fi first
#define se second
#define pii pair<int,int>
using namespace std;
const int INF = 0x3f3f3f3f;
typedef long long LL;
const int maxn = 1000+25;
int nextt[4][2] = {{0,1}, {1,0}, {0,-1}, {-1, 0}};
int n, m, G[maxn][maxn], A[maxn];
int fa[maxn*maxn];

int find_fa(int u){ return u == fa[u] ? u : fa[u] = find_fa(fa[u]); }

int main()
{
	freopen("in.txt","r",stdin);
	int T; scanf("%d",&T);
	while(T--){
		
		scanf("%d%d",&n,&m);
		for(int i = 1; i <= n; ++i) for(int j = 1; j <= m; ++j) scanf("%d", &G[i][j]);
		int s; scanf("%d",&s);
		for(int i = 1; i <= s; ++i) scanf("%d", &A[i]);
				
		for(int k = 1; k <= s; ++k){
			for(int i = 1; i <= n*m; ++i) fa[i] = i;	// 并查集初始化 
			for(int i = 1; i <= n; ++i){
				for(int j = 1; j <= m; ++j) if(G[i][j] > A[k]){
					int pa = find_fa((i-1)*n+j);
					// 4 个方向
					for(int p = 0; p < 4; ++p){
						int tx = i + nextt[p][0];
						int ty = j + nextt[p][1];
						if(tx > 0&&tx <= n&&ty > 0&&ty <= m&&G[tx][ty] > A[k]){
							int a1 = find_fa((tx-1)*n+ty);
							if(a1 == pa) continue;
							fa[a1] = pa;
						}
					}
				}
			}
			// 统计连通块
			int ans = 0;
			for(int i = 1; i <= n; ++i) for(int j = 1; j <= m; ++j){
				if(G[i][j] > A[k]&&(i-1)*n+j == find_fa((i-1)*n+j)){
					//printf("%d,%d ",i,j); 
					++ans;
				}
			}
			printf("%d ", ans); 
		}
		printf("\n");
	}

	return 0;
}

可以预料到，超时。
改进：
利用0 ≤ t1 ≤ t2 ≤···≤ tT 这一性质。
ti越小，联通块的总面积是越大的，而且是从ti+1的联通块发展过来的，即大于ti+1的一定大于ti。那么就倒序扫t数组，同时更新联通块。

然后问题来了，虽然我们知道ti+1用过就不用再考虑了，因为它必然大于ti，可是很难确定到底矩阵中那个数大于ti。我们可以对矩阵中的数排序，每次记住遍历到哪儿了，到了下一个ti，接着来遍历即可，这样整个矩阵中的数只用遍历1遍。

#include <cstdio>
#include <queue>
#include <vector>
#include <cstring>
#include <algorithm>
#define fi first
#define se second
#define pii pair<int,int>
using namespace std;
const int INF = 0x3f3f3f3f;
typedef long long LL;
const int maxn = 1000+25;
const int maxT = 100000 + 5;

int n, m, num[maxn][maxn];
int fa[maxn*maxn];
int A[maxT], Ans[maxT];

struct Point{
	int x,y,v;
	Point(int a, int b, int c):x(a),y(b),v(c){}
	bool operator < (const Point& rhs) const{
		return v > rhs.v;
	}
};
vector<Point> P;

// 找点 p 的邻接点 
void GetAdj(const Point& p, vector<int>& ans){
	ans.clear();
	int x = p.x - 1, y = p.y;
	if(x >= 0) ans.push_back(num[x][y]);
	x = p.x + 1;
	if(x < n) ans.push_back(num[x][y]);
	x = p.x; y = p.y - 1;
	if(y >= 0) ans.push_back(num[x][y]);
	y = p.y + 1;
	if(y < m) ans.push_back(num[x][y]);
}

int find_fa(int u){ return u == fa[u] ? u : fa[u] = find_fa(fa[u]); }

int main()
{
	//freopen("in.txt","r",stdin);
	int T; scanf("%d",&T);
	while(T--){
		P.clear();
		int cnt = 0;
		scanf("%d%d",&n,&m);
		for(int i = 0; i < n; ++i) for(int j = 0; j < m; ++j){
			int a; scanf("%d", &a);
			P.push_back(Point(i,j,a)); 
			++cnt;
		}
		
		sort(P.begin(), P.end());
		
		for(int i = 0; i < cnt; ++i){
			Point& p = P[i];
			fa[i] = i;
			num[p.x][p.y] = i;
		}
		int s; scanf("%d",&s);
		for(int i = 0; i < s; ++i) scanf("%d", &A[i]);
		
		int c = 0;
		Ans[s] = 0;
		for(int i = s-1; i >= 0; --i){
			int& ans = Ans[i];
			ans = Ans[i+1];
			vector<int> tmp;
			while(c < cnt&&P[c].v > A[i]){
				++ans;
				GetAdj(P[c], tmp);
				for(int j = 0; j < tmp.size(); ++j){
					Point& adj = P[tmp[j]];
					if(adj.v <= A[i]) continue;
					int adjfa = find_fa(tmp[j]);
					if(adjfa == find_fa(c)) continue;
					fa[adjfa] = c;
					--ans;
				}
				++c;
			}
		}
		
		for(int i = 0; i < s; ++i) printf("%d ", Ans[i]);
		printf("\n");
	}

	return 0;
}

*/