QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#570749#8790. First BillionHTensorTL 1007ms11132kbC++173.2kb2024-09-17 17:32:242024-09-17 17:32:24

Judging History

你现在查看的是最新测评结果

  • [2024-09-17 17:32:24]
  • 评测
  • 测评结果:TL
  • 用时:1007ms
  • 内存:11132kb
  • [2024-09-17 17:32:24]
  • 提交

answer

#include<iostream>
#include<vector>
#include<algorithm>
#include<random>
#include<array>
#include<cassert>
#define dd(x) cout << #x << "\n"
#define d(x) cout << #x  << ": " << x << "\n"
using namespace std;
#define int long long
using pii = pair<int, int>;
using vpii = vector<pii>;
using vi = vector<int>;
using vii = vector<vector<int>>;
using a3 = array<int, 3>;
using ll = long long;
// const int inf = 0x3f3f3f3f3f3f3f3fLL;

const int mod = 1e4;

void solve() {
    int n; cin >> n;
    vector<int> ori(n + 1);
    vector<pair<int, int>> a(n + 1);
    for(int i = 1; i <= n; i++) {
        cin >> ori[i];
        a[i].first = ori[i] % mod;
        a[i].second = i;
    }

    if(n == 2) {
        cout << (ll) 1e9 << "\n";
        return ;
    }

    vector<int> ans;
    mt19937 rng(19260817);
    auto check = [&]() {
        shuffle(a.begin() + 1, a.end(), rng);
        vector dp(n + 1, vector<int> (mod + 1));
        vector fr(n + 1, vector<pair<int, int>>(mod + 1));
        vector ch(n + 1, vector<int> (mod + 1));
        ans.clear();
        int sum = 0;

        auto positive = [](int x) {
            return (x % mod + mod) % mod;
        };

        for(int i = 1; i <= n; i++) {
            for(int j = 0; j < mod; j++) {
                int last = positive(j - a[i].first);
                if(dp[i - 1][j]) {
                    dp[i][j] = dp[i - 1][j];
                    fr[i][j] = fr[i - 1][j];
                    ch[i][j] = ch[i - 1][j];
                } else if(dp[i - 1][last]) {
                    dp[i][j] = dp[i - 1][last];
                    fr[i][j] = {i - 1, last};
                    ch[i][j] = i;
                }
            }

            if(dp[i][0]) {
                // cout << i << endl;
                auto [ii, ij] = pair(i, 0LL);
                // auto [ii, ij] = fr[i][0];
                while(pair(ii, ij) != pair(0LL, 0LL)) {
                    ans.push_back(a[ch[ii][ij]].second);
                    int nii = fr[ii][ij].first, nij = fr[ii][ij].second;
                    ii = nii, ij = nij;
                }
                break;
            }
            if(!dp[i][a[i].first]) {
                dp[i][a[i].first] = 1;
                ch[i][a[i].first] = i;
            }
        }

        for(auto pp : ans) {
            sum += ori[pp];
            // cout << pp << " ";
        }
        // cout << "\n";

        if(sum % mod) assert(false);

        if(sum != (long long)1e9) {
            return false;
        }
        return true;
    };

    while(!check()) {
        // assert(false);
        continue;
    }

    cout << ans.size() << " ";
    for(auto pp : ans) {
        cout << pp << " ";
    }
    cout << "\n";
}

signed main() {
    // ios::sync_with_stdio(false); cin.tie(0);
    int T = 1;
    while(T--) solve();
    return 0;
}

/*

10
386413329 88494216 245947398 316438989 192751270 204627269 65749456 3938400 150458676 345180997

24
17125795 281143405 10375259 196293002 158174864 34520650 52919232 87393970 99085271 62281508 67168428 55174991 54533464 51393059 89276370 41441658 72793517 30466999 73758332 97064918 111541434 142047546 12934221 101092107

*/

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 2ms
memory: 6760kb

input:

10
386413329 88494216 245947398 316438989 192751270 204627269 65749456 3938400 150458676 345180997

output:

5 1 9 7 5 6 

result:

ok OK (n = 10)

Test #2:

score: 0
Accepted
time: 0ms
memory: 6944kb

input:

10
119486233 299942886 169540407 349937991 597883752 32230162 140514533 57341098 12602102 220520836

output:

5 9 2 5 8 6 

result:

ok OK (n = 10)

Test #3:

score: 0
Accepted
time: 3ms
memory: 8076kb

input:

14
384615281 84612238 83310504 54746763 142296081 56775470 128760350 343006424 177232390 214368720 67220468 21895072 16352717 224807522

output:

7 12 9 1 13 10 7 6 

result:

ok OK (n = 14)

Test #4:

score: 0
Accepted
time: 3ms
memory: 7964kb

input:

14
270208635 14270307 89661499 113578022 47687195 101043954 38775146 208193324 650676076 351701957 3427619 59535626 24230888 27009752

output:

7 8 7 11 5 9 14 13 

result:

ok OK (n = 14)

Test #5:

score: 0
Accepted
time: 27ms
memory: 9876kb

input:

20
61638928 106712373 5946815 178135484 4937573 111395400 15504655 67139983 101814514 312223647 130341028 43244171 37671364 54108486 337181317 37924824 153793862 70383750 102917244 66984582

output:

10 6 3 7 19 15 1 17 2 16 20 

result:

ok OK (n = 20)

Test #6:

score: 0
Accepted
time: 114ms
memory: 10032kb

input:

20
67858098 61231428 99398662 1883806 82465954 303619377 87516412 154956240 94872199 76508350 13276828 136541811 203282099 99160366 127539385 13364660 141176136 39751629 67888657 127707903

output:

10 7 16 6 18 10 2 20 19 15 9 

result:

ok OK (n = 20)

Test #7:

score: 0
Accepted
time: 1007ms
memory: 11132kb

input:

24
17125795 281143405 10375259 196293002 158174864 34520650 52919232 87393970 99085271 62281508 67168428 55174991 54533464 51393059 89276370 41441658 72793517 30466999 73758332 97064918 111541434 142047546 12934221 101092107

output:

12 2 14 7 9 1 24 16 6 21 12 11 8 

result:

ok OK (n = 24)

Test #8:

score: -100
Time Limit Exceeded

input:

24
70224368 148769600 36654748 23404220 15009825 57449487 46896672 6065662 10377031 133719710 23220853 184445684 8462667 88501546 155244839 229323557 140109402 52520271 78995771 75721556 87987586 118427778 107013825 101453342

output:


result: