3048: [Usaco2013 Jan]Cow Lineup
Time Limit: 2 Sec Memory Limit: 128 MBSubmit: 225 Solved: 159
[Submit][Status][Discuss]
Description
Farmer John's N cows (1 <= N <= 100,000) are lined up in a row. Each cow is identified by an integer "breed ID" in the range 0...1,000,000,000; the breed ID of the ith cow in the lineup is B(i). Multiple cows can share the same breed ID. FJ thinks that his line of cows will look much more impressive if there is a large contiguous block of cows that all have the same breed ID. In order to create such a block, FJ chooses up to K breed IDs and removes from his lineup all the cows having those IDs. Please help FJ figure out the length of the largest consecutive block of cows with the same breed ID that he can create by doing this.
Input
* Line 1: Two space-separated integers: N and K.
* Lines 2..1+N: Line i+1 contains the breed ID B(i).
Output
* Line 1: The largest size of a contiguous block of cows with identical breed IDs that FJ can create.
Sample Input
2
7
3
7
7
3
7
5
7
INPUT DETAILS: There are 9 cows in the lineup, with breed IDs 2, 7, 3, 7, 7, 3, 7, 5, 7. FJ would like to remove up to 1 breed ID from this lineup.
Sample Output
OUTPUT DETAILS: By removing all cows with breed ID 3, the lineup reduces to 2, 7, 7, 7, 7, 5, 7. In this new lineup, there is a contiguous block of 4 cows with the same breed ID (7).
HINT
Source
尺取法
记录区间内每个元素出现次数和颜色种数
当种数<=k+1时r++,当种数>k+1时l++
1 #include<cstdio> 2 #include<iostream> 3 #include<algorithm> 4 #include<cstring> 5 #define N 100005 6 using namespace std; 7 int n,m,ans,num,a[N],b[N],cnt[N]; 8 int main(){ 9 scanf("%d%d",&n,&m); 10 for(int i=1;i<=n;i++) 11 scanf("%d",&a[i]),b[i]=a[i]; 12 sort(b+1,b+1+n); 13 int len=unique(b+1,b+1+n)-b-1; 14 for(int i=1;i<=n;i++) 15 a[i]=lower_bound(b+1,b+1+len,a[i])-b; 16 int l=1,r=1;cnt[a[1]]++;num++; 17 while(r<n){ 18 while(r<n&&num<=m+1){ 19 if(!cnt[a[++r]])num++; 20 cnt[a[r]]++; 21 ans=max(ans,cnt[a[r]]); 22 } 23 while(num>m+1&&l<=r) 24 if(!(--cnt[a[l++]]))num--; 25 } 26 printf("%d ",ans); 27 return 0; 28 }