poj 选择客栈

2019独角兽企业重金招聘Python工程师标准>>>

每找到一个符合消费条件的咖啡馆，进行一次不合法方式计数。
OAAAAAOAAAAOAAOOOAAAA
以O代表符合消费水平的咖啡馆（客栈），A代表某种颜色的客栈（这里所有A的颜色都是同种颜色）。以前7个客栈为例，两个O之前的5个客栈，任选两个都是不合法的选择方式，即C（5，2）种不合法的方式。
假设这里共有16个A颜色的客栈（注意这里的16包含O中颜色为A的个数），那么这里的总方法数为C（16，2），合法数为C(16,2) - C(5,2) - C(4,2) - C(2,2) - C(4,2).
对每种颜色分开考虑，再加起来就可以了。

/**
4G:选择客栈
查看 提交 统计 提问
总时间限制: 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。
*/

#include <cstdio>
#include <cstdlib>
using namespace std;

int n = 0, k = 0, p = 0;
int color[200005] = {0}, cost[200005] = {0};
int dp[200005][51] = {0};


int c(int m)
{
    return m * (m - 1) / 2;
}

int main()
{
    int sumn[51] = {0};
    int sum_temp[51] = {0};
    int sum_illegal = 0;//不合法方式总数
    int sum_total = 0;//总方法数
    scanf("%d%d%d", &n, &k, &p);
    //int mark = 0;
    for(int i = 0; i < n; ++i)
    {
        scanf("%d%d", &color[i], &cost[i]);
        sumn[color[i]]++;
        if(cost[i] <= p)
        {
            for(int j = 0; j < k; ++j)
            {
                sum_illegal += c(sum_temp[j]);
                sum_temp[j] = 0;
            }
        }
        else
        {
            sum_temp[color[i]]++;
        }
    }
    for(int j = 0; j < k; ++j)
        sum_illegal += c(sum_temp[j]);

    for(int j = 0; j < k; ++j)
    {
        sum_total += c(sumn[j]);
    }
    printf("%d\n", sum_total - sum_illegal);
    return 0;
}

转载于:https://my.oschina.net/locusxt/blog/140830