#NOIP2011#選擇客棧

【NOIP2011】選擇客棧

時間限制: 1 Sec  內存限制: 128 MB

題目描述

麗江河邊有 n 家很有特色的客棧，客棧按照其位置順序從1 到n 編號。每家客棧都按照某一種色調進行裝飾（總共k 種，用整數0 ~ k-1 表示），且每家客棧都設有一家咖啡店，每家咖啡店均有各自的最低消費。兩位遊客一起去麗江旅遊，他們喜歡相同的色調，又想嘗試兩個不同的客棧，因此決定分別住在色調相同的兩家客棧中。晚上，他們打算選擇一家咖啡店喝咖啡，要求咖啡店位於兩人住的兩家客棧之間（包括他們住的客棧），且咖啡店的最低消費不超過p。
他們想知道總共有多少種選擇住宿的方案，保證晚上可以找到一家最低消費不超過p元的咖啡店小聚。

輸入

輸入文件 hotel.in，共n+1 行。
第一行三個整數 n，k，p，每兩個整數之間用一個空格隔開，分別表示客棧的個數，色
調的數目和能接受的最低消費的最高值；
接下來的 n 行，第i+1 行兩個整數，之間用一個空格隔開，分別表示i 號客棧的裝飾色
調和i 號客棧的咖啡店的最低消費。

輸出

輸出文件名爲 hotel.out。輸出只有一行，一個整數，表示可選的住宿方案的總數。

樣例輸入

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

樣例輸出

3

記錄滿足要求的咖啡店的位置，同時枚舉右邊的客棧。

g[i][j]表示前i個客棧，顏色爲j的一共有多少個。

對於當前右邊的客棧i，它能夠貢獻的方案數爲，在它左邊,且最靠近它的咖啡店位置以左，與它顏色相同的客棧的個數。

如果當前客棧i滿足作爲咖啡店的價格要求，那麼它將貢獻g[i][Col[i]] - 1的方案數，即假設咖啡店在i處，與左邊的i - 1個客棧中顏色相同的個數。

Code:

#include<iostream> 
#include<cstdio> 
#include<cstdlib> 
#include<cstring> 
#include<algorithm> 
using namespace std; 
  
const int Max = 200000; 
const int Maxk = 50; 
  
int N, K, P; 
int g[Max + 5][Maxk + 5]; 
int Cnt[Maxk + 5], Cost[Max + 5], Col[Max + 5]; 
  
bool getint(int & num){ 
    char c; int flg = 1;    num = 0; 
    while((c = getchar()) < '0' || c > '9'){ 
        if(c == '-')   flg = -1; 
        if(c == -1) return 0; 
    } 
    while(c >= '0' && c <= '9'){ 
        num = num * 10 + c - 48; 
        if((c = getchar()) == -1)   return 0; 
    } 
    num *= flg; 
    return 1; 
} 
  
int main(){ 
    getint(N), getint(K), getint(P); 
    for(int i = 1; i <= N; ++ i) 
        getint(Col[i]), getint(Cost[i]), 
        ++ Cnt[Col[i]]; 
    for(int i = 1; i <= N; ++ i){ 
        for(int k = 0; k < K; ++ k) 
            g[i][k] = g[i - 1][k]; 
        ++ g[i][Col[i]]; 
    } 
    int Ans = 0, p = 1; 
    for(int i = 2; i <= N; ++ i){ 
        if(Cost[i] <= P)    p = i; 
        Ans += g[p][Col[i]] - (p == i ? 1 : 0); 
    } 
    printf("%d\n", Ans); 
    return 0; 
}