F. Polycarp and Hay (思维+bfs)

题目链接:F. Polycarp and Hay

题目大意:

给出一个n行m列的矩阵,一个k值,让按照要求修改矩阵

1.矩阵上的数值只能减小,不能增加
2.修改后的矩阵,一些位置上的点的大小要相等,其他位置上的点必须为0
3.这个位置上的点的和要等于k
4.这些点中至少有一个钥等于原矩阵中的对应位置的点的大小
5.这些点必须要联通
如果能得到这样的矩阵就输出,否则就输出NO;

思路:

1.从第四条规则切入,修改后的矩阵中肯定只包含原矩阵中的一个数字和0。
2.对矩阵每个点Ai j进行计算,如果A ij 的cnt倍等于k,并且矩阵中存在大于等于A ij并且和A ij联通的区域,该区域中的数字个数大小大于等于cnt,那么就是答案了。
3.用bfs求联通区域的大小,从i j开始搜索,然后查找和他联通的大小,查找到cnt个就可以返回ture。直接输出。

但是不做优化的话在95会TLE。每次搜索一个点,如果该点的cnt倍数等于k,但是它组成的联通块的大小小于cnt个,就无法构成答案,但是这些联通块中有和他相等的数字,那么这些数字也不用搜索,他们组成的联通块大小必然也无法大于cnt,这些就是无用的点。开个con数组,标记这些无需再搜索的点。

代码:

#include<iostream>
#include<vector>
#include<algorithm>
#include<cmath>
#include<string>
#include<queue>
#include<cstring> 
using namespace std;
const int N=1005;
int sk[N][N];
typedef long long ll;
typedef pair<int,int>Pii;
ll k;
vector<ll>v;
int sum=0,vis[N][N],n,m, con[N][N];
int dx[4]={0,1,0,-1},dy[4]={1,0,-1,0};
//查询联通块大小 
bool bfs(Pii p,int cnt)
{
    memset(vis,0,sizeof(vis));
    queue<Pii>q;
    q.push(p);
    vis[p.first][p.second]=1;
    int res=1;
    while(!q.empty())
    {
        if(res==cnt)return true;
        Pii pt=q.front();
        q.pop();
        int x=pt.first;
        int y=pt.second;
        for(int i=0;i<4;++i)
        {
            int nx=x+dx[i];
            int ny=y+dy[i];
            if(nx<0||nx>=n||ny<0||ny>=m)continue;
            if(!vis[nx][ny]&&sk[nx][ny]>=sk[p.first][p.second])
            {
                if(sk[nx][ny]==sk[p.first][p.second])con[nx][ny]=1;
                vis[nx][ny]=1;
                res++;
                q.push(Pii(nx,ny));
                if(res==cnt)return true;

            }
        }
    }
    return false;
}
int main()
{
    int flag=0;
    memset(con,0,sizeof(con));
    int i,j;
    cin>>n>>m>>k;
    for(i=0;i<n;++i)
    for(j=0;j<m;++j)
    scanf("%d",&sk[i][j]);
    int ans=-1;
    for(i=0;i<n;++i)
    {

    for(j=0;j<m;++j)
    {
        ll cnt=k/sk[i][j];
        //如果是该点的cnt倍数是k,并且没有被标记过无用点的就进行搜索。 
        if(cnt<=n*m&&!(k%sk[i][j])&&!con[i][j])
        {
            if(bfs(Pii(i,j),cnt))
            {
                ans=sk[i][j];
                flag=1;
                break;
            }
        }
    }
    if(flag)break;
    }
    if(ans==-1)
    {
        cout<<"NO";
        return 0;
    }
    cout<<"YES"<<endl;
    for(i=0;i<n;++i)
    for(j=0;j<m;++j)
    {
        if(vis[i][j])
        printf("%d%c",ans," \n"[j==m-1]);
        else printf("0%c"," \n"[j==m-1]);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/PinappleMi/article/details/80719653