Submission #46348

#	Time	Username	Problem	Language	Result	Execution time	Memory
46348		evpipis	Sirni (COCI17_sirni)	C++98	84 / 140	5117 ms	786432 KiB

sirni

#include <bits/stdc++.h>
using namespace std;

#define pb push_back
#define fi first
#define se second
#define mp make_pair
typedef pair<int, int> ii;
typedef long long ll;

const int len = 1e7+5;
int cnt[len], nex[2][len], p[len], r[len];
vector<ii> vec[len];

int fin(int i){
    if (p[i] == i) return i;
    return p[i] = fin(p[i]);
}

void join(int i, int j){
    if (r[i] > r[j])
        p[j] = i;
    else{
        p[i] = j;
        if (r[i] == r[j]) r[j]++;
    }
}

int main(){
    int n, m = 0, k = 0;
    scanf("%d", &n);
    for (int i = 0; i < n; i++){
        int temp;
        scanf("%d", &temp);
        m = max(m, temp);
        if (!cnt[temp]) cnt[temp] = ++k;
    }

    for (int i = m; i >= 1; i--){
        nex[0][i] = nex[0][i+1];
        if (cnt[i]) nex[0][i] = i;

        nex[1][i] = nex[1][i+1];
        if (cnt[i+1]) nex[1][i] = i+1;
    }

    for (int i = 1; i <= k; i++){
        r[i] = 0;
        p[i] = i;
    }

    for (int i = 1; i <= m; i++)
        if (cnt[i])
            for (int j = i; j <= m; j+=i){
                if (j == i && nex[1][j])
                    vec[nex[1][j]%i].pb(mp(cnt[i], cnt[nex[1][j]]));
                else if (j != i && nex[0][j])
                    vec[nex[0][j]%i].pb(mp(cnt[i], cnt[nex[0][j]]));
            }

    ll ans = 0;
    for (int i = 0; i <= m; i++)
        for (int j = 0; j < vec[i].size(); j++){
            int x = fin(vec[i][j].fi), y = fin(vec[i][j].se);
            if (x != y)
                join(x, y), ans += i;
        }

    printf("%lld\n", ans);
    return 0;
}

Compilation message (stderr)

sirni.cpp: In function 'int main()':
sirni.cpp:63:27: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (int j = 0; j < vec[i].size(); j++){
                         ~~^~~~~~~~~~~~~~~
sirni.cpp:31:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &n);
     ~~~~~^~~~~~~~~~
sirni.cpp:34:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d", &temp);
         ~~~~~^~~~~~~~~~~~~

• Subtask 1: 14 / 14
• Subtask 2: 0 / 14
• Subtask 3: 14 / 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