172702025-06-08 12:32:58algoproGamecpp17Time limit exceeded 30/1002.582s1092 KiB
// UUID: ecbc66b3-0be1-48b7-919f-3ee3fe69a45f
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

#include <iostream>

using namespace std;

const int MAX_N = 1e5;
int a[MAX_N];

int f[1 + MAX_N];

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int n, k;
    cin >> n >> k;
    for ( int i = 0; i < n; i ++ )
        cin >> a[i];
        
    for ( ; k > 0; k -- ) {
        int p;
        cin >> p;
        for ( int i = 1; i <= n; i ++ )
            f[i] = 0;
        for ( int i = 0; i < p; i ++ )
            f[a[i]] ++;
            
        int mx = n, pocket = 0, player = 1;
        long long score = 0;
        for ( int i = p; i < n; i ++, player *= -1 ) {
            if ( pocket ) {
                score += pocket * player;
                pocket = 0;
            } else {
                while ( !f[mx] )
                    mx --;
                f[mx] --;
                score += mx * player;
            }
            if ( a[i] <= mx )
                f[a[i]] ++;
            else
                pocket = a[i];
        }
        
        for ( int i = 0; i < p; i ++, player *= -1 ) {
            if ( pocket ) {
                score += pocket * player;
                pocket = 0;
            } else {
                while ( !f[mx] )
                    mx --;
                f[mx] --;
                score += mx * player;
            }
        }
        cout << score << '\n';
    }
    return 0;
}
SubtaskSumTestVerdictTimeMemory
subtask110/10
1Accepted1ms316 KiB
2Accepted1ms316 KiB
subtask220/20
1Accepted1ms316 KiB
2Accepted1ms316 KiB
3Accepted2ms316 KiB
4Accepted4ms436 KiB
subtask30/70
1Accepted28ms456 KiB
2Accepted30ms508 KiB
3Accepted104ms500 KiB
4Accepted126ms512 KiB
5Accepted527ms820 KiB
6Accepted1.047s856 KiB
7Accepted726ms820 KiB
8Accepted629ms1092 KiB
9Accepted998ms1076 KiB
10Accepted1.373s1080 KiB
11Accepted911ms1076 KiB
12Accepted2.411s1076 KiB
13Accepted1.583s1076 KiB
14Time limit exceeded2.582s1076 KiB