代码求调,5555

P1141 01迷宫

china_banana @ 2024-10-06 11:26:25

#include <bits/stdc++.h>
using namespace std;
int dx[5]={0,1,-1,0,0};
int dy[5]={0,0,0,1,-1};
int n,m;
struct node
{
    int x,y;
};
bool a[1005][1005];
void bfs(int startx,int starty)
{
    queue<node> q;
    bool v[1005][1005];
    q.push((node){startx,starty});
    v[startx][starty]=1;
    int tot=0;
    while(!q.empty())
    {
        node x=q.front();
        q.pop();
        for(int i=1;i<=4;i++)
        {
            int now_x=x.x+dx[i];
            int now_y=x.y+dy[i];
            if(now_x<1||now_x>n||now_y<1||now_y>n) continue;
            if(a[x.x][x.y]==a[now_x][now_y]||v[now_x][now_x]==1) continue;
            v[now_x][now_y]=1;
            tot++;
            q.push((node){now_x,now_y});
        }
    }
    printf("%d\n",tot);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            int x;
            scanf("%d",&x);
            if(x==1) a[i][j]==1;
            else a[i][j]==0;
        }
    }
    for(int i=1;i<=m;i++)
    {
        int q,e;
        scanf("%d %d",&q,&e);
        bfs(q,e);
    }
    return 0;
}

|