Submission #574878

#	Time	Username	Problem	Language	Result	Execution time	Memory
574878		talant117408	Bosses (BOI16_bosses)	C++17	100 / 100	677 ms	664 KiB

bosses

#include <bits/stdc++.h>
 
using namespace std;
 
typedef long long ll;
typedef pair <int, int> pii;
typedef pair <ll, ll> pll;
 
#define pb                  push_back
#define mp                  make_pair
#define all(v)              (v).begin(),(v).end()
#define sz(v)               int((v).size())
#define do_not_disturb      ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl                '\n'
 
int mod = 1e9+7;
 
ll modulo(ll a) {
    return ((a % mod) + mod) % mod;
}
 
ll add(ll a, ll b) {
    return modulo(a + b);
}
 
ll mult(ll a, ll b) {
    return modulo(a * b);
}
 
ll binpow(ll a, ll b) {
    ll res = 1;
    while (b) {
        if (b&1) {
            res = mult(res, a);
        }
        a = mult(a, a);
        b >>= 1;
    }
    return res;
}

ll bfs(int v, int n, vector <vector <int>> &graph) {
    vector <int> dist(n+1, 2e9);
    queue <int> K;
    dist[v] = 1;
    K.push(v);
    while (!K.empty()) {
        auto u = K.front(); K.pop();
        for (auto to : graph[u]) {
            if (dist[to] > dist[u] + 1) {
                dist[to] = dist[u] + 1;
                K.push(to);
            }
        }
    }
    
    ll total = 0;
    for (int i = 1; i <= n; i++) {
        if (dist[i] > n) return 9e18;
        total += dist[i];
    }
    return total;
}

void solve() {
    int n;
    cin >> n;
    vector <vector <int>> graph(n+1);
    for (int i = 1; i <= n; i++) {
        int k;
        cin >> k;
        for (int j = 0; j < k; j++) {
            int x;
            cin >> x;
            graph[x].pb(i);
        }
    }
    
    ll ans = 9e18;
    for (int i = 1; i <= n; i++) {
        ans = min(ans, bfs(i, n, graph));
    }
    cout << ans;
}

int main() {
    do_not_disturb
    
    int t = 1;
    //~ cin >> t;
    while (t--) {
        solve();
    }
    
    return 0;
}

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