Submission #1038183

#	Time	Username	Problem	Language	Result	Execution time	Memory
1038183		codexistent	Love Polygon (BOI18_polygon)	C++14	50 / 100	2071 ms	38500 KiB

polygon

#include <bits/stdc++.h>
using namespace std;
#define MAXN 100005
#define FOR(i, a, b) for(int i = a; i <= b; i++)
#define f first
#define s second
 
int n, ec[MAXN];
string st[MAXN], nx[MAXN];
map<string, int> idx;
set<int> adj[MAXN];
set<pair<int, int>> e;
bool d[MAXN];
 
int main(){
    ios::sync_with_stdio(false);
	cin.tie(nullptr);

    cin >> n;
    FOR(i, 1, n) ec[i] = 0;
    FOR(i, 1, n){
        cin >> st[i];
        idx.insert({st[i], i});
        cin >> nx[i];
    }
 
    if(n % 2) {
        cout << -1 << endl;
        return 0;
    }
 
    int r = 0;
    FOR(i, 1, n){
        d[i] = (idx[nx[idx[nx[i]]]] == i && idx[nx[i]] != i);
    }
 
    FOR(i, 1, n){
        if(i == idx[nx[i]] || d[i] || d[idx[nx[i]]]) continue;
        e.insert({i, idx[nx[i]]});
        ec[i]++, ec[idx[nx[i]]]++;
    }
 
    FOR(i, 1, n){
        if(i == idx[nx[i]] || d[i] || d[idx[nx[i]]]) continue;
        adj[i].insert(idx[nx[i]]);
        //cout << "DCVVB: " << idx[nx[i]] << endl;
        adj[idx[nx[i]]].insert(i);
    }
 
    set<pair<int, int>> se;
    FOR(i, 1, n) if(!d[i]) se.insert({ec[i], i});

    if(se.size() > n) {
        return 0;
    }
    
    while(se.size()){
        auto si = *se.begin();
 
        if(si.f == 0){
            r++;
            //cout << st[si.s] << " FALLALEUJAH! " << endl;
        }else{
            int j = 0, jmn = INT_MAX;
            for(int i : adj[si.s]){
                if(jmn > ec[i]) j = i, jmn = ec[i];
            }
            r++;
            //cout << st[si.s] << " ~ " << st[j] << " : " << si.s << " " << j << endl;
 
            se.erase({jmn, j});
            for(int j2 : adj[j]){
                se.erase({ec[j2], j2});
                adj[j2].erase(j), ec[j2]--;
                se.insert({ec[j2], j2});
            }
        }
 
        for(int i2 : adj[si.s]) {
            se.erase({ec[i2], i2});
            adj[i2].erase(si.s), ec[i2]--;
            se.insert({ec[i2], i2});
        }
 
        se.erase(se.begin());
    }

 
    cout << r << "\n";
}

Compilation message (stderr)

polygon.cpp: In function 'int main()':
polygon.cpp:53:18: warning: comparison of integer expressions of different signedness: 'std::set<std::pair<int, int> >::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   53 |     if(se.size() > n) {
      |        ~~~~~~~~~~^~~

• Subtask 1: 21 / 21
• Subtask 2: 0 / 25
• Subtask 3: 29 / 29
• Subtask 4: 0 / 25