Submission #532494

#	Time	Username	Problem	Language	Result	Execution time	Memory
532494		scottchou	Snowball (JOI21_ho_t2)	C++17	100 / 100	107 ms	13936 KiB

Main

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long LL;
int const N = 2e5 + 5;
LL a[N];
vector<LL> line, ans;
LL sum[N];
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n, q;
    cin >> n >> q;
    for(int i = 0; i < n; i++){
        cin >> a[i];
        if(i != 0){
            line.push_back(a[i] - a[i - 1]);
        }
    }
    sort(line.begin(), line.end());
    line.resize(unique(line.begin(), line.end()) - line.begin());
    ans.resize(line.size());
    LL nowx = 0, w, left = 0, right = 0, idx = 0;
    for(int i = 0; i < q; i++){
        cin >> w;
        nowx += w;
        if(nowx < 0)
            left = max(left, -nowx);
        else
            right = max(right, nowx);
        while(idx < line.size() && left + right >= line[idx]){
            if(w < 0){
                ans[idx] = right;
            }else{
                ans[idx] = line[idx] - left;
            }
            idx++;
        }
    }
    while(idx < line.size()){
        ans[idx] = -1;
        idx++;
    }
    sum[0] += left;
    sum[n - 1] += right;
    for(int i = 1; i < n; i++){
        LL len = a[i] - a[i - 1];
        LL lef = ans[lower_bound(line.begin(), line.end(), len) - line.begin()];
        if(lef == -1){
            sum[i - 1] += right;
            sum[i] += left;
            continue;
        }
        sum[i - 1] += lef;
        sum[i] += len - lef;
    }
    for(int i = 0; i < n; i++){
        cout << sum[i] << '\n';
    }
}

Compilation message (stderr)

Main.cpp: In function 'int main()':
Main.cpp:32:19: warning: comparison of integer expressions of different signedness: 'LL' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |         while(idx < line.size() && left + right >= line[idx]){
      |               ~~~~^~~~~~~~~~~~~
Main.cpp:41:15: warning: comparison of integer expressions of different signedness: 'LL' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   41 |     while(idx < line.size()){
      |           ~~~~^~~~~~~~~~~~~

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