Submission #948763

#	Time	Username	Problem	Language	Result	Execution time	Memory
948763		May27_th	Sirni (COCI17_sirni)	C++17	140 / 140	1867 ms	747088 KiB

sirni

#include<bits/stdc++.h>
#define taskname "a"
using namespace std;

void World_Final();
void Solve();

int main(){
    ios_base::sync_with_stdio(false); cin.tie(nullptr);
    if (fopen(taskname".in", "r")) {
        freopen(taskname".in", "r", stdin);
        freopen(taskname".out", "w", stdout);
    }
    World_Final();
}

void World_Final(){
    int Tests = 1;
    //cin >> Tests;
    for (int i = 1; i <= Tests; i ++) {
        //cout << i << "\n";
        Solve();
    }
}

struct DSU{
    private:
        vector<int> par, sz;
    public:
        int ccp;
        DSU(int N) : par(N + 5), sz(N + 5) {
            ccp = N;
            for (int i = 1; i <= N; i ++) {
                par[i] = i;
                sz[i] = 1;
            }
        }
        /*void init(int N) {
            par.resize(N + 5); sz.resize(N + 5);
            for (int i = 1; i <= N; i ++) {
                par[i] = i; sz[i] = 1;
            }
        }*/

        int findPar(int u) {
            return (u == par[u] ? u : par[u] = findPar(par[u]));
        }

        //int getSize(int u) { return sz[findPar(u)]; }
        bool same(int u, int v) {
            return findPar(u) == findPar(v);
        }

        bool unite(int u, int v) {
            int paru = findPar(u);
            int parv = findPar(v);
            if (paru == parv) return false;
            if (sz[paru] < sz[parv]) swap(paru, parv);
            sz[paru] += sz[parv];
            par[parv] = paru;

            return true;
        }

};

const int MAXN = 1e5 + 10;
const int64_t INF = 1e18;

struct Edge{
    int u, v, c;
};

void Solve(){
    int N; cin >> N;
    vector<int> P(N);
    for (auto &x : P) cin >> x;
    sort (P.begin(), P.end());
    P.erase(unique(P.begin(), P.end()), P.end());
    N = P.size();
    vector<int> pos(P.back() + 4, -1);
    for (int i = 0; i < N; i ++) {
        pos[P[i]] = i;
    }
    for (int i = P.back() - 1; i >= 0; i --) {
        if (pos[i] == -1) {
            pos[i] = pos[i + 1];
        }
    }
    vector<vector<pair<int, int>>> G(P.back() + 2);
    for (int i = 0; i < N - 1; i ++) {
        //ve.push_back({i, i + 1, min(P[i], P[i + 1] % P[i])});
        G[P[i + 1] % P[i]].push_back(make_pair(i, i + 1));
        for (int j = 2 * P[i]; j <= P.back(); j += P[i]) {
            //ve.push_back({i, pos[j], min(P[pos[j]] % P[i], P[i])});
            G[P[pos[j]] % P[i]].push_back(make_pair(i, pos[j]));
        }
    } /*sort (ve.begin(), ve.end(), [&] (Edge &a, Edge &b){
        return a.c < b.c;
    });*/

    int64_t ans = 0; DSU d(N + 10);
    for (int c = 0; c <= P.back(); c ++) {
        for (auto [i, j] : G[c]) {
            if (d.unite(i, j)) {
                ans += c;
            }
        }
    } cout << ans;


}

Compilation message (stderr)

sirni.cpp: In function 'int main()':
sirni.cpp:11:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   11 |         freopen(taskname".in", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~
sirni.cpp:12:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   12 |         freopen(taskname".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• 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