方法一:暴力枚举第一个客栈、第二个客栈以及两个客栈之间的客栈,寻找合法的条件,答案++。理论上也就30分吧,结果60分。。。贴一下代码
#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=200010;
int a[maxn],p[maxn];
int n,k,pr;
int ans=0;
int main()
{
cin>>n>>k>>pr;
for(int i=1;i<=n;i++)
{
cin>>a[i]>>p[i];
}
for(int i=1;i<=n;i++)
{
for(int j=i+1;j<=n;j++)
{
if(a[i]==a[j])
{
int s=i,t=j;
for(int kk=s;kk<=t;kk++)
{
if(p[kk]<=pr)
{
ans++;
break;
}
}
}
}
}
cout<<ans<<endl;
return 0;
}
方法二:递推
一种神奇的做法。我们假设tmp数组为某种颜色在读入那个客栈之前拥有的同色客栈数目,num数组代表某种颜色在读入那个客栈之前拥有的合法的同色客栈数目。循环1到n,读入color与price,如果当前的咖啡厅价格<=p,那么就把每种颜色的同色客栈的合法数目变成同色客栈数目(例如 红 红 红 此时tmp=3 合法的数目可能是2 读入一个price更低的红色客栈 则之前的所有红色客栈都合法了,num[j]=tmp[j]。当然,更新其他颜色是因为该客栈的咖啡厅价格低,对其他颜色的客栈更新也有贡献,存下来再说)这时更新答案+=num[color],num[color]++,tmp[color]++。如果读入的价格比p大,那么tmp[color]++,当前的答案更新(理由同上)
不太好理解 手动模拟一下吧。。。
#include<cstdio>
#include<iostream>
using namespace std;
int num[50];
int tmp[50];
int n,k,p;
long long ans=0;
int main()
{
int color,price;
cin>>n>>k>>p;
for(int i=1;i<=n;i++)
{
cin>>color>>price;
if(price<=p)
{
for(int j=0;j<k;j++) num[j]=tmp[j];
ans+=num[color];
num[color]++;
tmp[color]++;
}
else
{
tmp[color]++;
ans+=num[color];
}
}
cout<<ans<<endl;
return 0;
}