Submission #248349

#	Time	Username	Problem	Language	Result	Execution time	Memory
248349		VEGAnn	Sirni (COCI17_sirni)	C++14	56 / 140	5091 ms	786436 KiB

sirni

#include <bits/stdc++.h>
#define sz(x) ((int)x.size())
#define all(x) x.begin(),x.end()
#define PB push_back
#define i2 array<int,2>
using namespace std;
typedef long long ll;
const int N = 100100;
const int BIG = int(1e7) + 10;
const int cn = int(1e4);
const int oo = 2e9;
bool mrk[BIG];
vector<i2> edg;
vector<int> vc;
int n, nt[BIG];
ll ans = 0;

int get(int x) { return (x == nt[x] ? x : nt[x] = get(nt[x])); }

void calc(){
    for (int i = BIG - 2; i > 0; i--)
        if (mrk[i])
            nt[i] = i;
        else nt[i] = nt[i + 1];

    for (int cr : vc){
        int cm = mrk[cr];

        int x = cr;

        nt[get(x)] = get(x + 1);

        for (int i = get(x); i < BIG - 1; ){
            int nw = i % x;

            edg.PB({i, x});

            if (nw == 0)
                i += x;
            else i += x - nw;

            i = get(i);
        }
    }
}

bool cmp(i2 _x, i2 _y){
    return (_x[0] % _x[1]) < (_y[0] % _y[1]);
}

int main(){
    ios_base::sync_with_stdio(0); cin.tie(0);

#ifdef _LOCAL
    freopen("in.txt","r",stdin);
#endif // _LOCAL

    cin >> n;

    for (int i = 1; i <= n; i++){
        int x; cin >> x;
        mrk[x] = 1;
    }

    int siz = 0;

    for (int i = 1; i < BIG; i++)
        if (mrk[i]) {
            siz++;
            vc.PB(i);
        }

    nt[BIG - 1] = BIG - 1;

    calc();

    reverse(all(vc));

    calc();

    sort(all(edg), cmp);

    for (int cr : vc)
        nt[cr] = cr;

    for (i2 nw : edg){
        int x = get(nw[0]), y = get(nw[1]);

        if (x == y) continue;

        ans += nw[0] % nw[1];

        nt[x] = y;
    }

    cout << ans << '\n';

    return 0;
}

Compilation message (stderr)

sirni.cpp: In function 'void calc()':
sirni.cpp:27:13: warning: unused variable 'cm' [-Wunused-variable]
         int cm = mrk[cr];
             ^~

• Subtask 1: 0 / 14
• Subtask 2: 0 / 14
• Subtask 3: 0 / 14
• Subtask 4: 14 / 14
• Subtask 5: 14 / 14
• Subtask 6: 14 / 14
• Subtask 7: 14 / 14
• Subtask 8: 0 / 14
• Subtask 9: 0 / 14
• Subtask 10: 0 / 14