题目大意:
题目链接:https://www.luogu.org/problemnew/show/P1311
丽江河边有
n
n
n家很有特色的客栈,客栈按照其位置顺序从
1
1
1到
n
n
n编号。每家客栈都按照某一种色调进行装饰(总共
k
k
k种,用整数
0
∼
k
−
1
0\sim k-1
0∼k−1表示),且每家客栈都设有一家咖啡店,每家咖啡店均有各自的最低消费。
两位游客一起去丽江旅游,他们喜欢相同的色调,又想尝试两个不同的客栈,因此决定分别住在色调相同的两家客栈中。晚上,他们打算选择一家咖啡店喝咖啡,要求咖啡店位于两人住的两家客栈之间(包括他们住的客栈),且咖啡店的最低消费不超过
p
p
p 。
他们想知道总共有多少种选择住宿的方案,保证晚上可以找到一家最低消费不超过
p
p
p元的咖啡店小聚。
思路:
假设现在以
i
i
i为其中一个客栈,
1
∼
i
1\sim i
1∼i中最低消费不超过
p
p
p且尽量靠后的客栈为
j
j
j。那么明显任意一个
j
j
j左边的客栈且和
i
i
i颜色相同的客栈都可以和
i
i
i一起成为一个方案。
那么就从
1
1
1枚举到
n
n
n,每用
w
[
c
o
l
]
[
1
∼
s
u
m
]
w[col][1\sim sum]
w[col][1∼sum]表示
1
∼
i
1\sim i
1∼i中颜色是
c
o
l
col
col的客栈编号。而且保证
w
[
c
o
l
]
w[col]
w[col]单调递增。
记录
1
∼
i
1\sim i
1∼i中最靠后的最低消费不超过
p
p
p的客栈为
f
l
a
g
flag
flag,现在枚举到第
i
i
i个客栈,那么就在
w
[
c
o
l
[
i
]
]
w[col[i]]
w[col[i]]中二分,找到不超过
f
l
a
g
flag
flag的最靠后的客栈
j
j
j,那么
1
∼
j
1\sim j
1∼j都可以和
i
i
i匹配成为一种方案。
需要特判一下,如果客栈
i
i
i的最低消费已经没有超过
p
p
p了,那么
1
∼
i
−
1
1\sim i-1
1∼i−1的颜色相同的客栈都可以选择。
时间复杂度
O
(
n
l
o
g
n
)
O(n\ logn)
O(n logn)。
代码:
#include <cstdio>
#include <iostream>
using namespace std;
typedef long long ll;
const int N=200010;
int n,k,p,col[N],m[N],flag,w[60][N],sum[60],l,r,mid;
ll ans;
int main()
{
scanf("%d%d%d",&n,&k,&p);
for (int i=1;i<=n;i++)
scanf("%d%d",&col[i],&m[i]);
flag=-1;
for (int i=1;i<=n;i++)
{
if (m[i]<=p) ans+=(ll)sum[col[i]]; //特判
else
{
l=1;
r=sum[col[i]]; //表示颜色为i的客栈数量
while (l<=r)
{
mid=(l+r)/2;
if (flag>=w[col[i]][mid]) l=mid+1; //可以组合
else r=mid-1;
}
ans+=(ll)(l-1);
}
sum[col[i]]++; //数量加一
w[col[i]][sum[col[i]]]=i;
if (m[i]<=p) flag=i;
}
cout<<ans;
return 0;
}