求方案数至少有两类思路,一是建立递推关系用DP搞,二是用数据结构,把问题转化为一系列查询。
能不能把最低消费不超过p的咖啡店选出来,左边和右边同色的客栈数相乘,再加到一起,这样得到答案呢?会重复。怎样去重?不能把这种方案和以前出现过的方案做比较,这样就不能粗暴地乘起来了。可不可以多加一些限制呢?比如,这家咖啡店是所有合法咖啡店中最左边的一家?可不可以不在咖啡店上做文章,从客栈下手,比如,每对客栈只在左边这家统计一次?这样,问题就转化为n个查询:右边有多少家客栈和当前这家同色,并且两家客栈之间(含端点)有一家咖啡店的最低消费不超过p?这个查询有两个限制,是二维的。注意到颜色数比较少,可以直接用二维数组统计。友好的咖啡店可能有很多,我们需要知道最左边的那一家。有单调性,可以二分。更简便的方法是边算边更新。
每对客栈只在右边那家统计是一样的。以下代码空间复杂度可以优化。
#include <cstdio>
using namespace std;
const int MAX_N = 200000, MAX_K = 50;
int s[MAX_N+1][MAX_K];
int main()
{
int n, k, p, c, m, j = 0;
long long ans = 0;
scanf("%d %d %d", &n, &k, &p);
for (int i = 1; i <= n; ++i) {
scanf("%d %d", &c, &m);
if (m <= p)
j = i;
for (int l = 0; l < k; ++l)
s[i][l] = s[i-1][l];
ans += s[j][c];
++s[i][c];
}
printf("%lld\n", ans);
return 0;
}
要开学了……