Submission #199972

#	Time	Username	Problem	Language	Result	Execution time	Memory
199972		mdn2002	San (COCI17_san)	C++14	0 / 120	528 ms	65536 KiB

san

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
long long n,k,a[44],b[44],nm,tree[1200006],ans;
vector<pair<long long,long long> >on,to;
vector<long long>vt,tk;
map<long long,long long>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(y<l||r<x)return 0;
    int mid=(l+r)/2;
    return qr(nod*2,l,mid,x,y)+qr(nod*2+1,mid+1,r,x,y);
}
void bt(int x,int mx,long long sum)
{
    if(sum>=k)ans++;
    if(sum>0)
    {
        if(mx==n/2)on.push_back({a[x],sum});
        else
        {
            to.push_back({a[x],sum});
            tk.push_back(sum);
        }
        vt.push_back(sum);
    }
    for(int i=x+1;i<=mx;i++)
    {
        if(a[x]<=a[i])bt(i,mx,sum+b[i]);
    }
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    //freopen("gpsduel.in","r",stdin);
    //freopen("gpsduel.out","w",stdout);
    cin>>n>>k;
    for(int i=1;i<=n;i++)cin>>a[i]>>b[i];
    for(int i=1;i<=n/2;i++)bt(i,n/2,b[i]);
    for(int i=n/2+1;i<=n;i++)bt(i,n,b[i]);
    sort(on.begin(),on.end());
    sort(to.begin(),to.end());
    sort(vt.begin(),vt.end());
    sort(tk.begin(),tk.end());
    for(int i=0;i<vt.size();i++)
    {
        if(mp[vt[i]]==0)mp[vt[i]]=++nm;
    }
    int l=to.size()-1;
    for(int i=on.size()-1;i>=0;i--)
    {
        long long x=on[i].first,y=on[i].second;
        if(l!=-1)
        {
            while(to[l].first>=x)
            {
                long long sm=to[l].second;
                up(1,0,(1<<21)+45,mp[sm]);
                l--;
                if(l==-1)break;
            }
        }
        long long kk=*lower_bound(tk.begin(),tk.end(),k-y);
        if(lower_bound(tk.begin(),tk.end(),k-y)!=tk.end())
        {
            ans+=qr(1,0,(1<<21)+45,mp[kk],(1<<21)+45);
        }
    }
    cout<<ans;
}

Compilation message (stderr)

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

• Subtask 1: 0 / 24
• Subtask 2: 0 / 24
• Subtask 3: 0 / 24
• Subtask 4: 0 / 24
• Subtask 5: 0 / 24