QAQ
我的SB做法
把每个颜色的位置记下来,枚举一个开头颜色位置,看看后面最少到哪里就可以符合要求了,查询用线段树。(位置只会往后)
#include <cstdio>
#include <iostream>
#include <ctime>
using namespace std;
const int maxm=200100;
int p[maxm];
int st[maxm*4];
int wc[51][maxm];
int num[maxm];
void build(int o,int l,int r)
{
if(l==r)
{
st[o]=p[l];
return;
}
int mid=(l+r)/2;
build((o<<1),l,mid);
build((o<<1)|1,mid+1,r);
st[o]=min(st[(o<<1)],st[(o<<1)|1]);
}
int ask(int o,int l,int r,int ql,int qr)
{
if(ql<=l&&r<=qr)
return st[o];
if(r<ql||l>qr) return 99999999;
int mid=(l+r)/2;
//if(mid<ql) return ask((o<<1)|1,mid+1,r,ql,qr);
//if(mid>qr) return ask((o<<1),l,mid,ql,qr);
return min(ask((o<<1),l,mid,ql,qr),ask((o<<1)|1,mid+1,r,ql,qr));
}
int main()
{
//freopen("hotel.in","r",stdin);
//freopen("hotel.out","w",stdout);
int n,k,max_p;
scanf("%d%d%d",&n,&k,&max_p);
for(int i=1,x;i<=n;i++)
scanf("%d%d",&x,&p[i]),wc[x][++num[x]]=i;
build(1,1,n);
long long ans=0;
/*for(int i=1;i<=k;i++)
{
int l,r;
scanf("%d%d",&l,&r);
printf("%d ",ask(1,1,n,l,r));
}*/
/*for(int i=0;i<k;i++)
{
for(int j=1;j<=num[i];j++)
printf("%d ",wc[i][j]);
puts("");
}*/
for(int i=0;i<k;i++)
{
if(num[i]<=1) continue;
int f=2;
for(int j=1;j<num[i];j++)
{
int t;
for(t=max(f,j+1);t<=num[i];t++)
{
if(ask(1,1,n,wc[i][j],wc[i][t])<=max_p)
{
ans+=(num[i]-t+1);
break;
}
}
if(t==num[i]+1) break;
f=t;
}
}
printf("%lld",ans);
return 0;
}
牛逼的做法,在线做
tot[k]:记录之前颜色为k的客栈出现的次数;
sum[k]:记录之前“已找到小于花费”的颜色为k的客栈出现次数;
h[k]:当前颜色为k的客栈最后出现的位置;
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxm = 200000 + 5;
int k,s,h[maxm],ans,now;
int sum[maxm],tot[maxm];
int n;
int main()
{
int x,y;
scanf("%d%d%d",&n,&k,&s);
for(int i = 1;i <= n;++ i){
scanf("%d%d",&x,&y);
if(y<=s)
now = i;
if(now>=h[x])
sum[x] = tot[x];
h[x]=i;
tot[x] ++;
ans += sum[x];
}
cout<<ans;
return 0;
}