洛谷st表+贪心P1311 选择客栈

Posted 2020-11-04 DEVILK

【题目描述：】

丽江河边有n 家很有特色的客栈，客栈按照其位置顺序从 1 到n 编号。每家客栈都按照某一种色调进行装饰（总共 k 种，用整数 0 ~ k-1 表示），且每家客栈都设有一家咖啡店，每家咖啡店均有各自的最低消费。

两位游客一起去丽江旅游，他们喜欢相同的色调，又想尝试两个不同的客栈，因此决定分别住在色调相同的两家客栈中。晚上，他们打算选择一家咖啡店喝咖啡，要求咖啡店位于两人住的两家客栈之间（包括他们住的客栈），且咖啡店的最低消费不超过 p 。

他们想知道总共有多少种选择住宿的方案，保证晚上可以找到一家最低消费不超过 p元的咖啡店小聚。

【输入格式：】

输入文件hotel.in，共n+1 行。

第一行三个整数n ，k ，p，每两个整数之间用一个空格隔开，分别表示客栈的个数，色调的数目和能接受的最低消费的最高值；

接下来的n 行，第 i+1 行两个整数，之间用一个空格隔开，分别表示 i 号客栈的装饰色调和i 号客栈的咖啡店的最低消费。

【输出格式：】

输出文件名为hotel.out 。

输出只有一行，一个整数，表示可选的住宿方案的总数。

输入样例#1： 
5 2 3 
0 5 
1 3 
0 2 
1 4 
1 5 

输出样例#1： 
3

输入输出样例

 

【算法分析：】

正解是O(n)的复杂度，但2000000的数据nlog₂n就能过（2011年的时候自然是过不了的）

发现k种颜色分别是从0~k-1排的，可以把O(n³)暴力枚举的复杂度优化一下：

开一个二位数组c，c[i][j]表示第i种颜色的第j个客栈是c[i][j]号

循环开三层，第一层枚举颜色

第二、三层循环反向枚举这个颜色的客栈

第二层表示右边的客栈，第三层表示左边的客栈，

若是这两个客栈间的最小花费比p小（st表O(1)查询），则从第一个这个颜色的客栈开始一直到左边的这个客栈，都可以和右边的客栈匹配形成一组方案

效率大大提升.

 

【代码：】

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<algorithm>
 4 using namespace std;
 5 
 6 const int MAXN = 200000 + 10;
 7 const int K = 22;
 8 
 9 int n, k, p, ans;
10 int t[MAXN], c[51][MAXN], st[MAXN][K], Log2[MAXN];
11 struct hotel {
12     int color, cost, num;
13 }h[MAXN];
14 
15 inline int read() {
16     int x = 0; char ch = getchar();
17     while(ch < ‘0‘ || ch > ‘9‘) ch = getchar();
18     while(ch >= ‘0‘ && ch <= ‘9‘)
19         x = (x << 1) + (x << 3) + (ch - 48), ch = getchar();
20     return x;
21 }
22 
23 int Query(int l, int r) {
24     int x = Log2[r - l + 1];
25     return min(st[l][x], st[r - (1 << x) + 1][x]);
26 }
27 int main() {
28     n = read(), k = read(), p = read();
29     for(int i = 1; i <= n; i++) {
30         h[i].color = read(),
31         h[i].cost = read();
32         st[i][0] = h[i].cost;
33     }
34     for(int i = 1; i <= n; i++)
35         c[h[i].color][++t[h[i].color]] = i;
36     
37     for(int j = 1; j <= K; j++) {
38         for(int i = 1; i + (1 << j) - 1 <= n; i++)
39             st[i][j] = min(st[i][j - 1], st[i + (1 << (j - 1))][j - 1]);
40     }
41     for(int i = 0; (1 << i) <= n; i++)
42         Log2[1 << i] = i;
43     for(int i = 1; i <= n; i++)
44         if(!Log2[i]) Log2[i] = Log2[i - 1];
45         
46     for(int i = 0; i < k; i++) {
47         for(int end = t[i]; end >= 1; end--) {
48             int pos = 0;
49             for(int j = end - 1; j >= 1; j--) {
50                 if(Query(c[i][j], c[i][end]) <= p) {
51                     pos = j;
52                     break;
53                 }
54             }
55             ans += pos;
56         }
57     }
58     printf("%d\n", ans);
59 }