CODEVS1135选择客栈(2011noip提高组)

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

两位游客一起去丽江旅游,他们喜欢相同的色调,又想尝试两个不同的客栈,因此决定分别住在色调相同的两家客栈中。晚上,他们打算选择一家咖啡店喝咖啡,要求咖啡店位于两人住的两家客栈之间(包括他们住的客栈),且咖啡店的最低消费不超过p。
他们想知道总共有多少种选择住宿的方案,保证晚上可以找到一家最低消费不超过p元的咖啡店小聚。

思路:简单的组合搞定了这么一道看似。。。的题,很美丽啊!!!正难则反,结果就是所有的数减去不符合的数,注意每次的操作都是对同一种颜色进行的,最后c(n,2)(n个里面取2个的组合)=n*(n-1)/2。。。这个题目告诉我们,数学的思想要贯彻到OI里,OI也要贯彻到数学里。。。

code:

#include<iostream>
#include<cstdio>
using namespace std;
int sum[51]={0},summ[51]={0};
long long ans=0,agans=0;
int main()
{
int n,k,p,i,j,co,va;
cin>>n>>k>>p;
for (i=1;i<=n;++i)
{
  cin>>co>>va;
  ++co;
  ++sum[co];
  if (va<=p)
   for (j=1;j<=k;++j)
   {
     agans=agans+(summ[j]*(summ[j]-1))/2;
      summ[j]=0;
}
  else
    ++summ[co];
    } 
    for (i=1;i<=k;++i)
    {
     ans=ans+(sum[i]*(sum[i]-1))/2;
        agans=agans+(summ[i]*(summ[i]-1))/2;
}
cout<<ans-agans<<endl;

你可能感兴趣的:(code)