Codeforces Round #346 (Div. 2) F. Polycarp and Hay（并查集+bfs）

题目链接


思路：从大到小枚举矩阵里的每一个数，因为答案肯定是这其中的某一个数，假设当前数为x，然后把大于当前数的坐标放进一个集合（并查集实现），当并查集内的连通块个数大于k/x，那么就说明当前数X是可行方案，接着再bfs构造出k/x个x就好了，其他全为0.

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
const int dir[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
int n,m,father[maxn],a[1001][1001],vis[1001][1001],num[maxn];
struct node{
	int x,y,z;
};
vector<node>v;
bool cmp(const node &a,const node &b)
{
	return a.z>b.z;
}
int findfather(int x)
{
	if(x==father[x]) return x;
	int i=findfather(father[x]);
	father[x]=i;
	return i;
}
void unit(int a,int b)
{
	int fa=findfather(a),fb=findfather(b);
	if(fa!=fb)
	{
		father[fa]=fb;
		num[fb]+=num[fa],num[fa]=0;
	}
}
void bfs(int x,int y,ll cnt,int z)//t是个数，z是权值 
{
	queue<node>q;
	q.push({x,y,a[x][y]});
	vis[x][y]=1;cnt--;
	while(!q.empty())
	{
		node t=q.front();
		q.pop();
		int x=t.x,y=t.y,w=t.z;
		for(int j=0;j<4;++j)
		{
			if(cnt==0) break;
			int nx=x+dir[j][0],ny=y+dir[j][1];
			if(nx<1||nx>n||ny<1||ny>m||vis[nx][ny]) continue;
			if(a[nx][ny]>=z) vis[nx][ny]=1,q.push({nx,ny,a[nx][ny]}),cnt--;
		}
	}
	for(int i=1;i<=n;++i)
	for(int j=1;j<=m;++j)
	{
	if(vis[i][j]) printf("%d",z);
	else printf("0");
	printf("%s",j==m?"\n":" ");
	}
}
int main()
{
	ll k;
	for(int i=0;i<maxn;++i) father[i]=i,num[i]=1;
	scanf("%d%d%lld",&n,&m,&k);
	for(int i=1;i<=n;++i)
	for(int j=1;j<=m;++j)
	scanf("%d",&a[i][j]),v.push_back({i,j,a[i][j]});
	sort(v.begin(),v.end(),cmp);
	for(int i=0;i<v.size();++i)
	{
		int x=v[i].x,y=v[i].y,z=v[i].z;
		for(int j=0;j<4;++j)
		{
			int nx=x+dir[j][0],ny=y+dir[j][1];
			if(nx<1||nx>n||ny<1||ny>m) continue;
			if(a[nx][ny]>=z) unit((nx-1)*m+ny,(x-1)*m+y);
		}
		if(k%z!=0) continue;
		ll t=k/z;
		if(t<=num[findfather((x-1)*m+y)])
		{
			puts("YES");
			bfs(x,y,t,z);
			return 0;
		}
	}
	puts("NO");
}