Submission #61574307

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#include <bits/stdc++.h>
using namespace std;
#define ull unsigned long long
#define int long long
#define pii pair<int,int>
#define N 2000000
void solve() {
    int n;
    cin >> n;
    vector<int> a(n+9),s(N+9);
    auto add = [&](int x, int k) {
        for (; x < N; x+=x&-x) s[x]+=k;
    };
    auto query = [&](int x) {
        if (x == 0) return 0ll;
        int ans = 0;
        for (;x;x-=x&-x) ans+=s[x];
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

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

#define ull unsigned long long
#define int long long
#define pii pair<int,int>
#define N 2000000

void solve() {
    int n;
    cin >> n;
    vector<int> a(n+9),s(N+9);

    auto add = [&](int x, int k) {
        for (; x < N; x+=x&-x) s[x]+=k;
    };

    auto query = [&](int x) {
        if (x == 0) return 0ll;
        int ans = 0;
        for (;x;x-=x&-x) ans+=s[x];
        return ans;
    };

    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    
    for (int i = 1; i <= n; i++) {
        int ans = a[i]+query(N)-query(i-1);
        if (ans+i > 0 && ans+i < N) add(ans+i,1);
        cout << max(0ll,ans-(n-i)) << " ";
    }
}

signed main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int t = 1;
    for (int i = 0; i < t; i++) {
        solve();
    }
    return 0;
}

Submission Info

Judge Result