SCU 4487 king's trouble I

点击打开链接

暴力，直接遍历找一遍。

#include<bits/stdc++.h>
#define MOD 100000007
#define ll long long
const int maxn=4e2+40;
using namespace std;
int a[maxn][maxn];
int v[maxn][maxn];
int n,m,ans;
int dir[8][2]={{1,0},{0,1},{-1,0},{0,-1}};
int judge(int x,int y){
    if(x<1||x>n||y<1||y>m) return 0;
    if(v[x][y]) return 0;
    if(a[x][y]==0) return 0;
    return 1;
}
int dfs(int x,int y){
    int ret=0;
    for(int i=0;i<4;++i){
        int tx=x+dir[i][0];
        int ty=y+dir[i][1];
        if(!a[tx][ty]) ret++;
        if(!judge(tx,ty)) continue;
        v[tx][ty]=1;
        ret+=dfs(tx,ty);
    }
    return ret;
}
int main(){
    int i,T;
    scanf("%d",&T);
    while(T--){
        ans=0;memset(v,0,sizeof(v));memset(a,0,sizeof(a));
        scanf("%d %d",&n,&m);
        for(int i=1;i<=n;++i)
            for(int j=1;j<=m;++j)
                scanf("%d",&a[i][j]);
        for(int i=1;i<=n;++i)
            for(int j=1;j<=m;++j){
                if(a[i][j]==1&&v[i][j]==0) v[i][j]=1,ans+=dfs(i,j);
            }
        printf("%d\n",ans);
    }
    return 0;
}