제출 #1030966

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1030966		dpsaveslives	Bosses (BOI16_bosses)	C++17	0 / 100	1 ms	604 KiB

bosses

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 5010;
vector<int> adj[MAXN];
vector<int> graph[MAXN],st;
int N,ans;
int bfs(int start){
    queue<int> q; q.push(start);
    vector<int> dist(N+1,-1); dist[start] = 0;
    int maxd = -1;
    while(!q.empty()){
        int u = q.front(); q.pop();
        for(auto v : adj[u]){
            if(dist[v] == -1){
                dist[v] = dist[u]+1;
                q.push(v);
            }
        }
    }
    for(int i = 1;i<=N;++i){
        if(dist[i] == -1) return -1;
        maxd = max(maxd,dist[i]);
    }
    return maxd;
}
void dfs(int u, int p){
    for(auto v : graph[u]){
        if(v != p){
            dfs(v,u);
            st[u] += st[v];
        }
    }
    ++st[u];
    ans += st[u];
}
int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    cin >> N;
    for(int i = 1;i<=N;++i){
        int k; cin >> k;
        for(int j = 0;j<k;++j){
            int v; cin >> v;
            adj[v].push_back(i);
        }
    }
    int res = -1, root = -1;
    for(int i = 1;i<=N;++i){
        int ret = bfs(i);
        if(ret == -1) continue;
        if(ret < res || res == -1){
            res = ret;
            root = i;
        }
    }
    queue<int> q; q.push(root);
    //cout << root << "\n";
    vector<int> dist(N+1,-1),par(N+1,-1); dist[root] = 0;
    while(!q.empty()){
        int u = q.front(); q.pop();
        for(auto v : adj[u]){
            if(dist[v] == -1){
                dist[v] = dist[u]+1;
                par[v] = u;
                q.push(v);
            }
        }
    }
    st = vector<int>(N+1,0);
    for(int i = 1;i<=N;++i){
        if(par[i] == -1) continue;
        graph[par[i]].push_back(i);
        graph[i].push_back(par[i]);
    }
    ans = 0;
    dfs(root,0);
    cout << ans << "\n";
    return 0;
}

• Subtask 1: 0 / 22
• Subtask 2: 0 / 45
• Subtask 3: 0 / 33