Submission #701560

#	Time	Username	Problem	Language	Result	Execution time	Memory
701560		huomark1001	Snowball (JOI21_ho_t2)	C++17	100 / 100	105 ms	13784 KiB

Main

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

template<typename T> void _do(T x){cerr<<x<<"\n";}
template<typename T,typename ...U> void _do(T x,U ...y){cerr<<x<<", ";_do(y...);}
#define dbg(...) cerr<<#__VA_ARGS__<<" = ";_do(__VA_ARGS__);

#define float double
#define ss(n) fixed<<setprecision(n) 
#define ll long long
#define int ll
#define IO ios::sync_with_stdio(false);cin.tie(0);
#define ld long double
#define pb push_back
#define pii pair<int,int>
#define rep(i,a) for(int i=1;i<=a;i++)
#define rep0(i,a) for(int i=0;i<a;i++)
#define F first
#define S second
#define uni(a) a.resize(distance(a.begin(),unique(a.begin(),a.end())));
#define unisort(a) sort(a.begin(),a.end()); uni(a);

ll gcd(ll a,ll b){if(b==0) return a; return gcd(b,a%b);}

const ld pi=3.14159265358979323846264338327950288419716939931;
const int lar=3e18;
const int mol1=1e9+7;
const int mol2=998244353;
const int MX=2e5+5;


void solve()
{
    int n,q;
    cin>>n>>q;
    int a[n+1];int ans[n+1]={0};
    rep(i,n) cin>>a[i];
    int now=0;
    int mx[q+1]={0},mn[q+1]={0};
    int xx=0,n1=0,n2=0;
    rep(i,q){
        int g;
        cin>>g;
        xx+=g;
        n1=min(n1,xx);
        n2=max(n2,xx);
        mx[i]=max(mx[i],n2);
        mn[i]=min(mn[i],n1);
    }
    rep(i,n){
        if(i==1){
            int hh=mn[q];ans[1]+=abs(hh);
        }
        else{
            int l=0,r=q;int dif=a[i]-a[i-1];
            while(l<r){
                int mid=l+r+1>>1;
                if((mx[mid]+abs(mn[mid]))>=dif){
                    r=mid-1;
                }
                else l=mid;
            }
            int bb=l;
            int all=mx[bb]+abs(mn[bb]);
            ans[i-1]+=mx[bb];ans[i]+=abs(mn[bb]);
            if(bb!=q){
                if(mx[bb+1]>mx[bb]){
                    ans[i-1]+=dif-all;
                }
                else ans[i]+=dif-all;
            }
        }
    }
    ans[n]+=mx[q];
    rep(i,n) cout<<ans[i]<<"\n";

}

signed main()
{
    IO
    solve();
    return 0;
}

Compilation message (stderr)

Main.cpp: In function 'void solve()':
Main.cpp:58:28: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   58 |                 int mid=l+r+1>>1;
      |                         ~~~^~
Main.cpp:39:9: warning: unused variable 'now' [-Wunused-variable]
   39 |     int now=0;
      |         ^~~

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