oj.uz

답안 #888850

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
888850 2023-12-18T09:28:32 Z asdasdqwer Sirni (COCI17_sirni) C++14
98 / 140
 5000 ms 437800 KB 
#pragma GCC optimize("O3")
#pragma GCC target("avx,avx2,sse4")

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

struct DSU {
    vector<int32_t> p, r;
    DSU(int32_t n) {
        p.assign(n,0);
        r.assign(n,1);
        for (int i=0;i<n;i++)p[i]=i;
    }

    int get(int32_t i) {
        if (i != p[i]) p[i]=get(p[i]);
        return p[i];
    }

    bool unite(int32_t a,int32_t b) {
        // assert(a < p.size() && b < p.size());
        a=get(a);
        b=get(b);
        if (a==b)return false;
        if (r[a]<r[b])swap(a,b);
        p[b]=a;
        r[a]+=r[b];
        return true;
    }
};

signed test(int n, vector<int> b) {
    vector<int32_t> clos(b.back()+1,-1);
    for (int32_t x:b) {
        clos[x]=x;
    }

    for (int32_t i=clos.size()-2;i>=0;i--) {
        if (clos[i] == -1) {
            clos[i]=clos[i+1];
        }
    }

    // cout<<"here\n";

    typedef array<int32_t,3> pii;
    vector<pii> edg;
    for (int32_t x:b) {
        // cout<<x<<"\n";
        vector<pii> tmp;
        if (x != b.back()) {
            tmp.push_back({(clos[x+1] % x), x, clos[x+1]});
        }

        // cout<<"after\n";

        for (int32_t j=2*x;j<=b.back();j+=x) {
            // cout<<j<<" "<<clos[j]<<"\n";
            if (clos[j] != -1) {
                if (tmp.size() == 0) {
                    tmp.push_back({clos[j] % x, clos[j], x});
                }

                else if (tmp.back()[1] != clos[j]) {
                    tmp.push_back({clos[j] % x, clos[j], x});
                }
            }
        }

        if (tmp.size() == 0 || tmp.back()[1] != clos[b.back()]) {
            tmp.push_back({clos[b.back()] % x, clos[b.back()], x});
        }

        for (auto y:tmp) edg.push_back(y);
    }

    // cout<<edg.size()<<"\n";

    sort(edg.begin(),edg.end());

    map<int,int> mp;
    for (int i=0;i<b.size();i++) {
        mp[b[i]]=i;
    }

    for (auto &x:edg) {
        // cout<<x[1]<<" "<<x[2]<<"\n";
        x[1]=mp[x[1]];
        x[2]=mp[x[2]];
    }
    // cout<<"\n";

    DSU d(b.size());
    int cost=0;
    for (auto x:edg) {
        if (d.unite(x[1], x[2])) {
            // cout<<b[x[1]]<<" "<<b[x[2]]<<"\n";
            cost+=x[0];
        }
    }
    return cost;
}

bool brute() {
    int len=(rand() % 100000)+100000;
    vector<int> b;
    set<int> chose;
    for (int i=0;i<len;i++) {
        int j=(rand()%100000)+1;
        while (chose.find(j) != chose.end()) {
            j=(rand()%100000)+1;
        }

        chose.insert(j);
        b.push_back(j);
    }

    cout<<"gen numbers\n";

    sort(b.begin(),b.end());
    typedef array<int,3> pii;
    vector<pii> g;
    for (int i=0;i<len;i++) {
        for (int j=i+1;j<len;j++) {
            g.push_back({b[j] % b[i], j, i});
        }
    }

    sort(g.begin(),g.end());

    DSU d(len);
    int cost=0;
    for (auto x:g) {
        if (d.unite(x[1], x[2])) {
            // cout<<b[x[1]]<<" "<<b[x[2]]<<"\n";
            cost += x[0];
        }
    }

    cout<<"mst calc\n";

    // cout<<"\n";
    int cost2 = test(len, b);
    cout<<"mst fast\n";
    // cout<<"\n";
    if (cost != cost2) {
        cout<<cost<<" "<<cost2<<"\n";
    }
    // for (int x:b) {
    //     cout<<x<<" ";
    // }
    // cout<<"\n";
    cout<<"finished\n";
    cout.flush();
    return true;
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int n;cin>>n;
    vector<int> a(n);
    for (int &x:a)cin>>x;
    if (n == 1) {
        cout<<0<<"\n";
        return 0;
    }
    sort(a.begin(),a.end());
    vector<int> b;
    for (int x:a) {
        if (b.size()==0||b.back()!=x)b.push_back(x);
    }

    cout<<test(n,b)<<"\n";
}

Compilation message

sirni.cpp: In function 'int test(int, std::vector<int>)':
sirni.cpp:82:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   82 |     for (int i=0;i<b.size();i++) {
      |                  ~^~~~~~~~~

Subtask #1
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 29 ms 39584 KB Output is correct
2 Correct 87 ms 43528 KB Output is correct
3 Correct 30 ms 39684 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 672 KB Output is correct
2 Correct 242 ms 40424 KB Output is correct
3 Correct 30 ms 39760 KB Output is correct

Subtask #3
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 29 ms 39604 KB Output is correct
2 Correct 28 ms 39512 KB Output is correct
3 Correct 29 ms 39772 KB Output is correct

Subtask #4
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 371 ms 32588 KB Output is correct
2 Correct 1202 ms 57496 KB Output is correct
3 Correct 477 ms 32668 KB Output is correct

Subtask #5
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 48 ms 7780 KB Output is correct
2 Correct 647 ms 55820 KB Output is correct
3 Correct 352 ms 29176 KB Output is correct

Subtask #6
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 779 ms 55708 KB Output is correct
2 Correct 1567 ms 107228 KB Output is correct
3 Correct 461 ms 32412 KB Output is correct

Subtask #7
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 114 ms 9036 KB Output is correct
2 Correct 1630 ms 105692 KB Output is correct
3 Correct 449 ms 31752 KB Output is correct

Subtask #8
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 515 ms 68416 KB Output is correct
2 Execution timed out 5045 ms 437800 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #9
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 506 ms 68712 KB Output is correct
2 Execution timed out 5049 ms 435784 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #10
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 96 ms 43532 KB Output is correct
2 Execution timed out 5066 ms 436776 KB Time limit exceeded
3 Halted 0 ms 0 KB -