流感传染

- 看完题目，发现时间即使是O(n^3)也就可以应付，那就直接愉快的扫吧

#include<cmath>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cstdlib>
#include<algorithm>
using namespace std;

int b[100][100];
char a[100][100];

void f(int n)
{
    int i,j;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            if(a[i][j]=='@'&&b[i][j]==1){
                if(i-1>=0&&a[i-1][j]=='.') a[i-1][j]='@';
                if(i+1<n&&a[i+1][j]=='.') a[i+1][j]='@';
                if(j-1>=0&&a[i][j-1]=='.') a[i][j-1]='@';
                if(j+1<n&&a[i][j+1]=='.') a[i][j+1]='@';
            }
}
int main()
{
    char str[101];
    int n,i,j,m,count=0;
    scanf("%d",&n);
    getchar();
    memset(b,0,sizeof(int));
    for(i=0;i<n;i++){
        gets(str);
        for(j=0;j<n;j++){
            a[i][j]=str[j];
            if(a[i][j]=='@'){
                b[i][j]=1;
            }
        }
    }
    scanf("%d",&m);
    m=m-1;
    while(m--){
        f(n);
        for(i=0;i<n;i++){
            for(j=0;j<n;j++){
                if(a[i][j]=='@'&&b[i][j]==0){
                    b[i][j]=1;
                }
            }
        }
    }
    for(i=0;i<n;i++){
        for(j=0;j<n;j++){
            if(a[i][j]=='@'){
                count++;
            }
        }
    }
    printf("%d",count);
    return 0;
}