PAT 甲级 1021 Deepest Root【DFS+BFS】

这道题没啥花里胡哨,就是基本的图(或者说树)的遍历。
因为N是10^4,如果用邻接矩阵的话可能会超时,所以这里用的邻接表的存储方式。
用dfs查连通分量数
用bfs计算最深的根层数
bfs计算层数使用的方法是在每一层的后面将一个特殊点入队作为层的分隔点,在本题中这个点就是0,因为结点都是大于0的。

ps:我又来偷懒了,用set存结点就不用再排序了

//DFS+BFS

const int maxn = 10000 + 5;

int n;
int v[maxn];
vector<int> G[maxn];
set<int> ans;
queue<int> q;

void dfs(int x){
    v[x] = 1;
    int t = G[x].size();
    for(int i=0;i<t;i++){
        if(!v[G[x][i]]) dfs(G[x][i]);
    }
}

//因为没有0结点,所以借用0作为层之间的分界点
//每层后面跟个0用来分层
int bfs(int i){
    int deep = 0;

    while(!q.empty()) q.pop();
    memset(v,0,sizeof(v));

    q.push(i);
    q.push(0);
    while(!q.empty()){
        int r = q.front();
        q.pop();
        if(r == 0){
            if(q.empty()) break;
            deep++;
            q.push(0);
            continue;
        }
        v[r] = 1;

        for(int i=0;i<G[r].size();++i){
            if(!v[G[r][i]]){
                q.push(G[r][i]);
            }
        }


    }

    return deep;
}

void process(){
    ans.clear();
    int maxx = 0;
    for(int i=1;i<=n;++i){
        int t = bfs(i);
        if(maxx > t) continue;
        else if(maxx == t) ans.insert(i);
        else {
            ans.clear();
            ans.insert(i);
            maxx = t;
        }
    }

    for(set<int>::iterator it = ans.begin();it!=ans.end();++it)
        printf("%d\n",*it);
}

int main()
{
    int x,y,e,con;
    scanf("%d",&n);
    for(int i=0;i<n-1;i++){
        scanf("%d%d",&x,&y);
        G[x].push_back(y);
        G[y].push_back(x);
    }
    memset(v,0,sizeof(v));
    e = 0;
    con = 0;
    for(int i=1;i<=n;i++){
        if(!v[i]){
            dfs(i);
            con++;
        }
    }
    if(con>1) printf("Error: %d components\n",con);
    else process();
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章