树的宽度 XJOI

树的宽度

http://115.236.49.52:83/problem/9521

题目描述：

已知一棵树，有n个结点，编号1至N，其中1号是根。求树的宽度

输入格式：

第一行一个数N（1<=N<=1000）

接下来N行每行N个1或0，第i第j列是1，表示i,j两点有边，否是没有边

输出格式：

树的宽度

样例输入：

10
0110000000
1001000000
1000110001
0100000000
0010000000
0010001000
0000010110
0000001000
0000001000
0010000000

样列输出：

4

题意

给一棵树，求其宽度

CODE

#include <cstdio>
#include <iostream>
using namespace std;

int n, tot;
string s[1010];
bool a[1010][1010], b[1010];
int ans[1010];

void dfs(int k, int deep){
    if(b[k]){//父亲节点，不用去
        return ;
    }
    ans[deep]++;//节点数++
    b[k] = true;//标记
    if(ans[deep] > tot){
        tot = ans[deep];
    }
    for(int i = 1; i <= n; i++){//扫一遍
        if(a[i][k]){
            dfs(i, deep + 1);
        }
    }
    return ;
}

int main(){
    scanf("%d", &n);
    for(int i = 1; i <= n; i++){
        cin >> s[i];
    }
    for(int i = 1; i <= n; i++){//存
        for(int j = 1; j <= n; j++){
            if(s[i][j-1] == '1'){
                a[i][j] = a[j][i] = 1;
            }
        }
    }
    dfs(1, 1);
    printf("%d\n", tot);
    return 0;
}