题目链接
思路:从大到小枚举矩阵里的每一个数,因为答案肯定是这其中的某一个数,假设当前数为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");
}