对于一个符合要求的点 在其两侧找到两个相同颜色客栈
我的第一反应 分治。。 (虽然可以dp 可以线段树。。)
分治一类最好使用递归实现
比较方便 递归写的时候只管当前的状态(普遍性) 再设置边界 就好了
readdata时记录下符合要求 即价格合适的点的位置 以及数量
int rec[200000+10];//价格合适的点的位置 由小到大
int cnt;//记录个数
在预处理前缀和
我的第一反应 分治。。 (虽然可以dp 可以线段树。。)
分治一类最好使用递归实现
比较方便 递归写的时候只管当前的状态(普遍性) 再设置边界 就好了
readdata时记录下符合要求 即价格合适的点的位置 以及数量
int rec[200000+10];//价格合适的点的位置 由小到大
int cnt;//记录个数
在预处理前缀和
int now[200000+10][51];//截至第i格位置 为j颜色的有多少个客栈
友情提示 极限情况会爆int 切勿作死
and windows系统下long long 为I64d Linux系统下为lld
友情提示 极限情况会爆int 切勿作死
and windows系统下long long 为I64d Linux系统下为lld
分治部分伪代码
void divide(左右标记)
{
if边界
{
特判
返回
}
递归左
递归右
for颜色
ans+=左区间同颜色数*右区间同颜色数;
ans--//防止 分界点本身的自己选择自己
}
输出ans就对了O(∩_∩)O~
void divide(左右标记)
{
if边界
{
特判
返回
}
递归左
递归右
for颜色
ans+=左区间同颜色数*右区间同颜色数;
ans--//防止 分界点本身的自己选择自己
}
输出ans就对了O(∩_∩)O~
/*类nlogn算法 找到一个价格合适的点 分治左右 */
#include<iostream>
#include<cstdio>
using namespace std;
int n,k,p;
int now[200000+10][51];//截至第i格位置 为j颜色的有多少个客栈
int rec[200000+10];//价格合适的点的位置 由小到大
int cnt;//记录个数
struct edge
{
int col;
int pri;
}node[200000+10];
long long ans;
void divide(int lp,int rp,int l,int r) //一段区间里开头的分界点 lp 结尾分界点rp
{
int midp=(lp+rp)/2;
if(lp>rp)
return;
if(lp==rp)
{
if(l==r)
return;
for(int i=0;i<=k;i++)
ans+=(now[rec[midp]][i]-now[l-1][i])*(now[r][i]-now[rec[midp]-1][i]);
ans--;
return;
}
divide(lp,midp-1,l,rec[midp]-1);
divide(midp+1,rp,rec[midp]+1,r);
for(int i=0;i<=k;i++)
ans+=(now[rec[midp]][i]-now[l-1][i])*(now[r][i]-now[rec[midp]-1][i]);
ans--;
}
void readdata()
{
scanf("%d%d%d",&n,&k,&p);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&node[i].col,&node[i].pri);
if(node[i].pri<=p)
rec[++cnt]=i;
for(int j=0;j<=k;j++)
{
now[i][j]=now[i-1][j];
if(node[i].col==j)
now[i][j]++;
}
}
}
int main()
{
freopen("hotel.in","r",stdin);
freopen("hotel.out","w",stdout);
readdata();
divide(1,cnt,1,n);
printf("%I64d\n",ans);
}