Submission #407190

#	Time	Username	Problem	Language	Result	Execution time	Memory
407190		Aldas25	Teleporters (IOI08_teleporters)	C++14	85 / 100	647 ms	65540 KiB

teleporters

#pragma GCC optimize("O2")

#include <bits/stdc++.h>

using namespace std;

#define FAST_IO ios_base::sync_with_stdio(0); cin.tie(0)
#define FOR(i, a, b) for(int i = (a); i <= (b); i++)
#define REP(n) FOR(O, 1, (n))
#define pb push_back
#define f first
#define s second
typedef long double ld;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<int, pii> piii;
typedef vector<int> vi;
typedef vector<pii> vii;
typedef vector<ll> vl;
typedef vector<piii> viii;

const int MAXN = 2000100, MAXK = 30;
//const ll MOD = 998244353;
const ll INF = 1e17;
const ld PI = asin(1) * 2;

int n, m;
int w[MAXN], e[MAXN];
vi seq;
unordered_map<int, int> toId;
int to[MAXN];
vi lens;

bool vis[MAXN];
int len = 0;
void dfs (int v) {
    vis[v] = true;
    len++;
    if (!vis[to[v]])
        dfs(to[v]);
}

int main()
{
    FAST_IO;

    cin >> n >> m;
    FOR(i, 1, n) cin >> w[i] >> e[i];
    FOR(i, 1, n) {
        seq.pb(w[i]);
        seq.pb(e[i]);
    }
    sort(seq.begin(), seq.end());

    int curId = 1;
    for (int x : seq)
        toId[x] = ++curId;

    seq.clear();

    FOR(i, 1, n) {
        int id1 = toId[w[i]];
        int id2 = toId[e[i]];
        to[id1-1] = id2;
        to[id2-1] = id1;
      //  cout << "  i = " << i << " w = " << w[i] << " e = " << e[i] << " id1 = " << id1 << " id2= " << id2 << endl;
    }

    toId.clear();

    //FOR(i, 1, 2*n+1) cout << " i = " << i << " to = " << to[i] << endl;

    vis[0] = true;

    ll ans = 0;
    FOR(i, 1, 2*n+1) {
        if (vis[i]) continue;
        len = 0;
        dfs(i);
        if (i == 1) {
            len--;
            ans = len;
        } else {
            lens.pb(len);
        }
    }
   // cout << "ans = " << ans << "\n";

    sort(lens.begin(), lens.end());
    reverse(lens.begin(), lens.end());

    for (int x : lens) {
        if (m == 0) break;
        m--;
        ans += x + 2;
      //  cout << " adding x+2;  x=" << x << endl;
    }

    int kk = m/2;
    m%=2;
    ans += 4*kk;
    if (m == 1) ans ++;

    cout << ans << "\n";

    return 0;
}

• Subtask 1: 5 / 5
• Subtask 2: 5 / 5
• Subtask 3: 5 / 5
• Subtask 4: 5 / 5
• Subtask 5: 5 / 5
• Subtask 6: 5 / 5
• Subtask 7: 5 / 5
• Subtask 8: 5 / 5
• Subtask 9: 5 / 5
• Subtask 10: 5 / 5
• Subtask 11: 5 / 5
• Subtask 12: 5 / 5
• Subtask 13: 5 / 5
• Subtask 14: 5 / 5
• Subtask 15: 5 / 5
• Subtask 16: 5 / 5
• Subtask 17: 5 / 5
• Subtask 18: 0 / 5
• Subtask 19: 0 / 5
• Subtask 20: 0 / 5