제출 #201790

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
201790		mdn2002	Vudu (COCI15_vudu)	C++14	42 / 140	946 ms	65540 KiB

vudu

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
long long n,p,a[1000006],sum[1000006],tree[8000006];
vector<long long>pr;
map<long long,int>mp;
void up(int nod,int l,int r,int x)
{
    if(x<l||r<x)return;
    if(l==r)
    {
        tree[nod]++;
        return;
    }
    int mid=(l+r)/2;
    up(nod*2,l,mid,x);
    up(nod*2+1,mid+1,r,x);
    tree[nod]=tree[nod*2]+tree[nod*2+1];
}
long long qr(int nod,int l,int r,int x,int y)
{
    if(x<=l&&r<=y)return tree[nod];
    if(r<x||y<l)return 0;
    int mid=(l+r)/2;
    return qr(nod*2,l,mid,x,y)+qr(nod*2+1,mid+1,r,x,y);
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    //freopen("empty.in","r",stdin);
    //freopen("empty.out","w",stdout);
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
        sum[i]=sum[i-1]+a[i];
    }
    cin>>p;
    for(int i=0;i<n;i++)
    {
        long long x=p-sum[i]+(p*i);
        long long y=-sum[i-1]+(p*i);
        pr.push_back(x);
        pr.push_back(y);
    }
    sort(pr.begin(),pr.end());
    int cnt=1;
    for(int i=0;i<pr.size();i++)
    {
        if(mp[pr[i]]==0)mp[pr[i]]=cnt++;
    }
    long long ans=0;
    for(int i=0;i<n;i++)
    {
        long long x=mp[p-sum[i]+(p*i)];
        long long y=mp[-sum[i-1]+(p*i)];
        up(1,0,n*2+55,y);
        ans+=qr(1,0,n*2+55,x,n*2+55);
    }
    cout<<ans;
}

컴파일 시 표준 에러 (stderr) 메시지

vudu.cpp: In function 'int main()':
vudu.cpp:50:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<pr.size();i++)
                 ~^~~~~~~~~~

• Subtask 1: 42 / 140