題目描述 Description
麗江河邊有 n 家很有特色的客棧,客棧按照其位置順序從1 到n 編號。每家客棧都按照某一種色調進行裝飾(總共k 種,用整數0 ~ k-1 表示),且每家客棧都設有一家咖啡店,每家咖啡店均有各自的最低消費。
兩位遊客一起去麗江旅遊,他們喜歡相同的色調,又想嘗試兩個不同的客棧,因此決定分別住在色調相同的兩家客棧中。晚上,他們打算選擇一家咖啡店喝咖啡,要求咖啡店位於兩人住的兩家客棧之間(包括他們住的客棧),且咖啡店的最低消費不超過p。
他們想知道總共有多少種選擇住宿的方案,保證晚上可以找到一家最低消費不超過p元的咖啡店小聚。
輸入描述 Input Description
共n+1 行。
第一行三個整數 n,k,p,每兩個整數之間用一個空格隔開,分別表示客棧的個數,色調的數目和能接受的最低消費的最高值;
接下來的 n 行,第i+1 行兩個整數,之間用一個空格隔開,分別表示i 號客棧的裝飾色調和i 號客棧的咖啡店的最低消費。
輸出描述 Output Description
輸出只有一行,一個整數,表示可選的住宿方案的總數。
範例輸入 Sample Input
5 2 3
0 5
1 3
0 2
1 4
1 5
範例輸出 Sample Output
3
資料範圍及提示 Data Size & Hint
【輸入輸出範例說明】
客棧編號 ① ② ③ ④ ⑤
色調 0 1 0 1 1
最低消費 5 3 2 4 5
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。
60分的做法:直接暴力類比。
#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int maxn=200000+20;int n,k,p;struct node{ int c,w;}a[maxn];int main(){ scanf("%d%d%d",&n,&k,&p); for(int i=1;i<=n;i++) { scanf("%d%d",&a[i].c,&a[i].w); } int ans=0; for(int i=1;i<=n;i++) { for(int j=i+1;j<=n;j++) if(a[i].c==a[j].c) { for(int k=i;k<=j;k++) if(a[k].w<=p) { ans++; break; } } } printf("%d\n",ans); return 0;}
正解:依然是類比
記錄一個c[k]數組表示第k種色調最後出現的位置,tot[k]記錄之前第k種色調出現的總次數,sum[k]表示可以加入當前答案的,從c[k]到當前客棧的顏色為k的客棧數量。
邊讀入邊處理,如果當前客棧咖啡店i的最小花費≤要求的最低花費,則令h=i,暫時記錄最低花費≤p的最右邊的位置。如果這個位置在已經確定的位置後面,則之前所有顏色為k的客棧都可以加入答案,sum=tot。
#include<iostream>#include<cstdio>using namespace std;const int maxn=200000+2;int sum[maxn],tot[maxn],c[maxn];int n,k,p,h,ans;int main(){ int x,y; scanf("%d%d%d",&n,&k,&p); for(int i=1;i<=n;i++) { scanf("%d%d",&x,&y); if(y<=p) h=i; if(h>=c[x]) sum[x]=tot[x]; c[x]=i; ans+=sum[x]; tot[x]++; } cout<<ans<<'\n'; return 0;}