open judge选择客栈 解题记录

总时间限制: 

1000ms 

内存限制: 

65535kB

描述

丽江河边有 n 家很有特色的客栈，客栈按照其位置顺序从 1 到n 编号。每家客栈都按照某一种色调进行装饰（总共 k 种，用整数 0 ~ k-1 表示） ，且每家客栈都设有一家咖啡店，每家咖啡店均有各自的最低消费。 
两位游客一起去丽江旅游，他们喜欢相同的色调，又想尝试两个不同的客栈，因此决定分别住在色调相同的两家客栈中。晚上，他们打算选择一家咖啡店喝咖啡，要求咖啡店位于两人住的两家客栈之间（包括他们住的客栈） ，且咖啡店的最低消费不超过 p。 
他们想知道总共有多少种选择住宿的方案，保证晚上可以找到一家最低消费不超过 p 元的咖啡店小聚。

输入

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

输出

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

样例输入

5 2 3 
0 5 
1 3 
0 2 
1 4 
1 5 

样例输出

3 

提示

2 人要住同样色调的客栈，所有可选的住宿方案包括：住客栈①③，②④，②⑤，④⑤，
但是若选择住 4、5 号客栈的话，4、5 号客栈之间的咖啡店的最低消费是 4，而两人能承受的最低消费是 3 元，所以不满足要求。因此只有前 3 种方案可选。 


【数据范围】 
对于 30%的数据，有 n ≤ 100； 
对于 50%的数据，有 n ≤ 1,000； 

对于 100%的数据，有 2 ≤ n≤ 200,000，0 < k ≤ 50，0 ≤ p ≤ 100， 0 ≤ 最低消费 ≤ 100。

题目的分析：

这道题目本质上就是一道数学题。想像在这条街上符合两个人要求的咖啡店是一些标记，这些标记把这条街分成了很多段。在第i段里面的各个颜色的客栈的数目分别是val[i].num[0]~val[i].num[k-1]。对于每个标记也把它看作是单独的段。

在这些实现了之后就可以开始计算了，就以颜色j为例。设所有颜色为j的客栈共有sum个，那么在不考虑咖啡店的情况下一共有 sum*(sum-1)/2种选择方式。在考虑咖啡店之后，只需要减去在同一条线段上的选择方法为(val[i].num[j]-1)*val[i].num[j]/2个。将所有的做一个运算之后就得到了总的结果。

下面是代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
struct line
{
	int num[50];
	int s;
	line() :s(0) { memset(num, 0, sizeof(num)); }
};
line temp;
int main()
{
	int n, k, p;
	cin >> n >> k >> p;
	vector<line>val;
	int tmp_colour;
	int tmp_p;
	for (int i = 0; i < n; i++) {
		cin >> tmp_colour >> tmp_p;
		if (tmp_p <= p) {
			if (temp.s != 0)
				val.push_back(temp);
			memset(temp.num, 0, sizeof(temp.num));
			temp.s = 1;
			temp.num[tmp_colour] = 1;
			val.push_back(temp);
			temp.s = 0;
			memset(temp.num, 0, sizeof(temp.num));
			continue;
		}
		temp.s = 1;
		temp.num[tmp_colour]++;
	}
	if (temp.s != 0)
		val.push_back(temp);
	int solution = 0;
	int sum = 0; int tmp = 0;
	int length = val.size();
	for (int j = 0; j < k; j++) {
		sum = 0; tmp = 0;
		for (int i = 0; i < length; i++) {
			int d = val[i].num[j];
			sum += d;
			tmp += (d*(d - 1) / 2);
		}
		int f = sum *(sum - 1) / 2 - tmp;
		solution += f;
	}
	cout << solution << endl;
	return 0;
}

不好的地方可能在于使用了STL库，会略微拖慢一点速度QAQ

总体来讲，只要理清了思路，这道题是不困难的