码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2018-05-10 21:39:50      阅读:170      评论:0      收藏:0      [点我收藏+]

标签:分享   三层   AC   枚举   main   图片   二层   cst   多少   

【题目描述:】

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

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

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

【输入格式:】

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

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

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

【输出格式:】

输出文件名为hotel.out 。

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

技术分享图片
输入样例#15 2 3 
0 5 
1 3 
0 2 
1 4 
1 5 

输出样例#13
输入输出样例

 

【算法分析:】

正解是O(n)的复杂度,但2000000的数据nlog2n就能过(2011年的时候自然是过不了的)

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

开一个二位数组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 }

 

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

标签:分享   三层   AC   枚举   main   图片   二层   cst   多少   

原文地址:https://www.cnblogs.com/devilk-sjj/p/9021645.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!