Submission #954597

#	Time	Username	Problem	Language	Result	Execution time	Memory
954597		HuyAT	Sirni (COCI17_sirni)	C++14	140 / 140	4304 ms	648224 KiB

sirni

#include<bits/stdc++.h>

const int MaxV = 1e7 + 10;

struct DSU{
    std::vector<int> lab;
    DSU(int n) : lab(n + 1,-1){

    }
    int findSet(int i){
        return (lab[i] < 0 ? i : lab[i] = findSet(lab[i]));
    }
    bool sameSet(int x,int y){
        return (findSet(x) == findSet(y));
    }
    void merg(int x,int y){
        x = findSet(x);
        y = findSet(y);
        if(x == y){
            return;
        }
        if(-lab[x] > -lab[y]){
            std::swap(x,y);
        }
        lab[y] += lab[x];
        lab[x] = y;
    }
};

bool mark[MaxV + 1];
int n,nxt[MaxV + 1];
std::vector<std::pair<int,int>> edge[MaxV + 1];

void readData(){
    std::cin >> n;
    for(int i = 1;i <= n;++i){
        int v;
        std::cin >> v;
        mark[v] = true;
    }
}
void precompute(){
    int t = -1;
    for(int i = MaxV;i >= 1;--i){
        if(!mark[i]){
            nxt[i] = t;
            continue;
        }
        if(t != -1){
            edge[t % i].emplace_back(i,t);
        }
        t = i;
        nxt[i] = i;
        for(int j = i * 2;j <= MaxV;j += i){
            if(mark[j] && i != j){
                edge[0].emplace_back(i,j);
                continue;
            }
            if(nxt[j] != -1){
                j = (nxt[j] / i) * i;
                edge[nxt[j] - j].emplace_back(i,nxt[j]);
            }else{
                break;
            }
        }
    }
}
long long solve(){
    long long ans = 0;
    DSU graph(MaxV);
    for(int i = 0;i <= MaxV;++i){
        for(std::pair<int,int> e : edge[i]){
            if(!graph.sameSet(e.first,e.second)){
                ans += i;
                graph.merg(e.first,e.second);
            }
        }
    }
    return ans;
}
int main(){
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);std::cout.tie(nullptr);
    readData();
    precompute();
    std::cout << solve();
    return 0;
}

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