Submission #935083

#	Time	Username	Problem	Language	Result	Execution time	Memory
935083		haxorman	Snowball (JOI21_ho_t2)	C++14	100 / 100	86 ms	14272 KiB

Main

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

#define int long long

const int mxN = 2e5 + 7;

int arr[mxN], ans[mxN];

int32_t main() {
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    
    int n, q;
    cin >> n >> q;
    
    vector<pair<int,int>> dist;
    for (int i = 0; i < n; ++i) {
        cin >> arr[i];

        if (i && abs(arr[i] - arr[i-1])) {
            dist.push_back({abs(arr[i] - arr[i-1]), i-1});
        }
    }
    sort(dist.begin(), dist.end(), greater<pair<int,int>>());

    int l = 0, r = 0, cur = 0, val = INT_MIN;
    for (int i = 0; i < q; ++i) {
        int w;
        cin >> w;

        cur += w;
        l = min(l, cur);
        r = max(r, cur);
    
        while (dist.size() && abs(l) + r >= dist.back().first) {
            auto [d, id] = dist.back();
            dist.pop_back();

            //cout << d << ' ' << id << ' ' << l << ' ' << r << ' ' << w << '\n';

            if (w>=0) {
                ans[id] += d-abs(l);
                ans[id+1] += abs(l);
            }
            else {
                ans[id+1] += d-r;
                ans[id] += r;
            }
        }
    }

    ans[0] += abs(l), ans[n-1] += r;
    while (dist.size()) {
        auto [d, id] = dist.back();
        dist.pop_back();
        
        ans[id] += r, ans[id+1] += abs(l);
    }
    
    for (int i = 0; i < n; ++i) {
        cout << ans[i] << "\n";
    }
}

Compilation message (stderr)

Main.cpp: In function 'int32_t main()':
Main.cpp:36:18: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   36 |             auto [d, id] = dist.back();
      |                  ^
Main.cpp:54:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   54 |         auto [d, id] = dist.back();
      |              ^
Main.cpp:26:32: warning: unused variable 'val' [-Wunused-variable]
   26 |     int l = 0, r = 0, cur = 0, val = INT_MIN;
      |                                ^~~

• Subtask 1: 33 / 33
• Subtask 2: 67 / 67