oj.uz

Submission #948750

# Submission time Handle Problem Language Result Execution time Memory
948750 2024-03-18T13:04:50 Z May27_th Sirni (COCI17_sirni) C++17
84 / 140
 972 ms 786432 KB 
#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<Edge> ve;
    int mx = 1e7 + 10;
    vector<int> P(N), pos(mx, -1);
    for (auto &x : P) cin >> x;
    sort (P.begin(), P.end());
    P.erase(unique(P.begin(), P.end()), P.end());
    N = P.size();
    for (int i = 0; i < N; i ++) {
        pos[P[i]] = i;
    }
    for (int i = 10000000; i >= 1; i --) {
        if (pos[i] == -1) {
            pos[i] = pos[i + 1];
        }
    }
    for (int i = 0; i < N; i ++) {
        if (i < N - 1) {
            ve.push_back({i, i + 1, min(P[i], P[i + 1] % P[i])});
        }
        for (int j = 2 * P[i]; j <= mx - 10; j += P[i]) {
            if (pos[j] == -1) break;
            ve.push_back({i, pos[j], min(P[pos[j]] % P[i], P[i])});
        }
    } sort (ve.begin(), ve.end(), [&] (Edge &a, Edge &b){
        return a.c < b.c;
    });
    int64_t ans = 0; DSU d(N);
    for (auto [u, v, c] : ve) {
        if (d.unite(u, v)) {
            ans += c;
        }
    } cout << ans;


}

Compilation message

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.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 28 ms 39512 KB Output is correct
2 Correct 171 ms 89076 KB Output is correct
3 Correct 29 ms 39964 KB Output is correct

Subtask #2
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 28 ms 39768 KB Output is correct
2 Runtime error 735 ms 786432 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #3
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 28 ms 39616 KB Output is correct
2 Correct 28 ms 39516 KB Output is correct
3 Correct 28 ms 39868 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 95 ms 66192 KB Output is correct
2 Correct 241 ms 90020 KB Output is correct
3 Correct 142 ms 90780 KB Output is correct

Subtask #5
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 39 ms 42708 KB Output is correct
2 Correct 185 ms 89752 KB Output is correct
3 Correct 91 ms 64960 KB Output is correct

Subtask #6
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 158 ms 89508 KB Output is correct
2 Correct 325 ms 140224 KB Output is correct
3 Correct 127 ms 66728 KB Output is correct

Subtask #7
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 50 ms 46032 KB Output is correct
2 Correct 363 ms 139896 KB Output is correct
3 Correct 129 ms 65216 KB Output is correct

Subtask #8
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 129 ms 64704 KB Output is correct
2 Runtime error 759 ms 786432 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #9
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 149 ms 64840 KB Output is correct
2 Runtime error 748 ms 786432 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #10
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 45 ms 42692 KB Output is correct
2 Runtime error 972 ms 786432 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -