Submission #374964

#	Time	Username	Problem	Language	Result	Execution time	Memory
374964		gustason	Fire drill (LMIO18_sauga)	C++14	31.46 / 100	60 ms	5868 KiB

sauga

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
struct Node {
    int cnt, in, idx;
    bool operator<(Node other) {
        if (in + cnt == other.in + cnt)
            return idx < other.idx;
        return in + cnt < other.in + cnt;
    }
};
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    srand(time(0));
    int t, n, S;
    cin >> t >> n >> S;

    vector<int> cant[n+1], in[n+1];
    for(int i = 0; i < n; i++) {
        int m;
        cin >> m;
        for(int j = 0; j < m; j++) {
            int x;
            cin >> x;
            cant[i].push_back(x);
            in[x].push_back(i);
        }
    }

    set<pair<int, int>> s;
    vector<int> cnt(n+1), inn(n+1);
    for(int i = 0; i < n; i++) {
        s.insert({cant[i].size(), i});
        cnt[i] = cant[i].size();
        inn[i] = in[i].size();
    }

    vector<bool> done(n+1, false);
    vector<int> ans;

    for(int i = 0; i < n; i++) {
        int best = n+1, best2 = 0, idx = 0;
        for(int j = 0; j < n; j++) {
            if (done[j]) continue;
//            cout << cnt[j] << " " << inn[j] << "\n";
            if (cnt[j] < best || (cnt[j] == best && inn[j] > best2)) {
                best = cnt[j];
                best2 = inn[j];
                idx = j;
            }
        }
        cerr << "\n";

        done[idx] = true;
        ans.push_back(idx);

        for(int j : in[idx]) {
            cnt[j]--;
        }
        for(int j : cant[idx]) {
            inn[j]--;
        }
//        auto it = s.begin();
//        while(1) {
//            it = s.begin();
//            if (done[(*it).second]) {
//                s.erase(it);
//                continue;
//            }
//            break;
//        }
//        done[(*it).second] = true;
//        ans.push_back((*it).second);
//        s.erase(it);
//        for(int j : in[(*it).second]) {
//            s.erase({cnt[j], j});
//            cnt[j]--;
//            s.insert({cnt[j], j});
//        }
    }

    for(int i : ans) {
        cout << i + 1 << "\n";
    }
    return 0;
}

• Subtask 1: 31.458438956200002 / 100