题目(以加强版为主)
题目描述
丽江河边有 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 元的咖啡店小聚。
输入格式
输入共 n + 1 n+1 n+1 行。
第一行三个整数 n , k , p n,k,p n,k,p,每两个整数之间用一个空格隔开,分别表示客栈的个数,色调的数目和能接受的最低消费的最高值。
接下来的 n n n 行,第 i + 1 i+1 i+1 行两个整数,之间用一个空格隔开,分别表示 i i i 号客栈的装饰色调和 i i i 号客栈的咖啡店的最低消费。
输出格式
输出一行一个整数,表示可选的住宿方案的总数。
样例 #1
样例输入 #1
5 2 3
0 5
1 3
0 2
1 4
1 5
样例输出 #1
3
提示
【样例解释】
客栈编号 ① ② ③ ④ ⑤ 色调 0 1 0 1 1 最低消费 5 3 2 4 5 \def\arraystretch{1.5} \begin{array}{|c|c|c|c|c|c|}\hline \textsf{客栈编号} & \text{①} & \text{②} & \text{③} & \text{④} & \text{⑤} \\\hline \textsf{色调} & 0 & 1 & 0 & 1 & 1 \\\hline \textsf{最低消费} & 5 & 3 & 2 & 4 & 5 \\ \hline \end{array} 客栈编号色调最低消费①05②13③02④14⑤15
二人要住同样色调的客栈,所有可选的住宿方案包括:住客栈①③,②④,②⑤,④⑤。
但是若选择住 ④⑤ 号客栈的话,④⑤ 号客栈之间的咖啡店的最低消费是 4 4 4,而两人能承受的最低消费是 3 3 3 元,所以不满足要求。因此只有前 3 3 3 种方案可选。
【数据范围】
对于
25
%
25\%
25% 的数据,
n
≤
100
n\leq 100
n≤100;
对于
40
%
40\%
40% 的数据,
n
≤
1000
n\leq 1000
n≤1000;
对于
80
%
80\%
80% 的数据,
n
≤
2
×
1
0
5
n\leq 2 \times 10^5
n≤2×105,
k
≤
50
k \leq 50
k≤50;
对于
100
%
100\%
100% 的数据,
2
≤
n
≤
2
×
1
0
6
2\leq n\leq2\times 10^6
2≤n≤2×106,
1
≤
k
≤
1
0
4
1 \le k\leq 10^4
1≤k≤104,
0
≤
p
≤
100
0\leq p\leq 100
0≤p≤100,
0
≤
0\leq
0≤ 最低消费
≤
100
\leq 100
≤100 。
题目分析及解答
普通版
直接暴力( 70 % 70\% 70%)
考虑到这道题大多数点的数据不大,可以直接暴力枚举所以客栈选择并进行判断,可以拿到大多数分数。
#include <iostream>
using namespace std;
int n,k,p,m[200300][2],cnt=0;
int main()
{
cin>>n>>k>>p;
for(int i=1;i<=n;i++)
{
cin>>m[i][0]>>m[i][1];//输入颜色与最低消费
}
for(int i=1;i<=n;i++)
{
for(int j=i+1;j<=n;j++)
{
if(m[i][0]!=m[j][0]) continue;//如果颜色不同,不判断
for(int k=i;k<=j;k++)
if(m[k][1]<=p)//若存在任意一个咖啡店符合要求,答案加一
{
cnt++;
break;
}
}
}
cout<<cnt;
return 0;
}
优化暴力( 100 % 100\% 100%)
先前那种暴力会在判断颜色是否相同时耗过多时间。因此,我们可以分别存每种颜色所对应的客栈位置,再遍历两个相同颜色客栈之间咖啡馆的最低消费。(其实这样做仍旧可能过不了一些情况,比如说颜色全都相同时。但因为这道题的数据相当水,所以仍旧能够满分过,那些会被卡的点要用加强版代码解决。)
#include <iostream>
using namespace std;
int n,k,p,m[200300][2],c[51][200300],cnt=0;
int main()
{
cin>>n>>k>>p;
for(int i=1;i<=n;i++)
{
cin>>m[i][0]>>m[i][1];
c[m[i][0]][++c[m[i][0]][0]]=i;//录入对应颜色数组
}
for(int i=0;i<k;i++)//遍历颜色
{
for(int j=1;j<=c[i][0];j++)
{
for(int l=j+1;l<=c[i][0];l++)
for(int f=c[i][j];f<=c[i][l];f++)
if(m[f][1]<=p)
{
cnt++;
break;
}
}
}
cout<<cnt;
return 0;
}
加强版
分析
对于每个点,如果自身消费小于最大值,那么它就可以使自身以及之前的所有点(不管颜色是否相同)与其之后的同色点组合。因此,我们可以对于每个点进行判断,如果消费小于最大值,就使自身及自身前未被标记的点被被标记,并将答案加上同色被标记点数目减一(不能和自身组合),否则将答案加同色被标记点数。
栗子
以本题样例为栗:
对于点一,消费大于最大值,不标记,答案加同色标记点数量
=
0
=0
=0;
对于点二,消费等于最大值,标记自身与点一,答案加同色标记点数量
−
1
=
0
-1=0
−1=0;
对于点三,消费小于最大值,标记自身,答案加同色标记点数量
−
1
=
1
-1=1
−1=1;
对于点四,消费大于最大值,不标记,答案加同色标记点数量
=
2
=2
=2;
对于点五,消费大于最大值,不标记,答案加同色标记点数量
=
3
=3
=3;
代码实现
#include <iostream>
using namespace std;
long long n,k,p,sum[10050],cnt=0;
struct node{
int c,v;
}m[2000400];
int main()
{
cin>>n>>k>>p;
for(int i=1;i<=n;i++)
{
cin>>m[i].c>>m[i].v;
}
int j=0;//一开始标记到第0个点
for(int i=1;i<=n;i++)
{
if(m[i].v<=p)
{
while(j<i)
{
j++;//更新目前标记到的点编号
sum[m[j].c]++;
}
cnt+=sum[m[i].c]-1;//注意:自己不能和自己组合,故要减一
}
else cnt+=sum[m[i].c];//这里因为没有标记自己而不用考虑和自己组合的情况
}
cout<<cnt;
return 0;
}
注意事项
对于加强版,数据组数与答案大小会变大,记得改数组大小并开 l o n g long long l o n g long long。