题目描述
“蓝猫淘气三千问,看蓝猫,我有姿势我自豪!”话说能考上HYSBZ的孩纸们肯定都是很有姿势的孩纸们,但是大家普遍偏科,都只有一门科目考得好。已知HYSBZ的入学考试科目数量小于等于109,而有n个学生参加了入学考试。现在HYSBZ要刷人了,招生办每一次刷人会把一个科目考得好的人全部刷掉,但是最多不能刷超过K次。(刷就是不录取)而HYSBZ的校长看录取名单时,最喜欢看的就是连续都是同一个科目考得好的人。他定义完美学生序列为连续且考得好的科目都为同一门的学生序列。现在招生办主任想让你帮他设计一种录取方案,使得最长的完美学生序列尽量长。
输入
共N+1行,第一行2个正整数n,K,n表示入学考试人数,K表示刷人次数上限。
接下来N行,每行仅一个正整数Ai,为i号学生所考得好的科目。
输出
仅1个正整数,为最长的最长完美学生序列。
样例输入
9 1
2
7
3
7
7
3
7
5
7
样例输出
4
提示
[样例解释]
总共有9个学生,最多只能刷一次学生。
若不刷,最长完美学生序列长度为2
若刷掉考第3门考得好的学生,则学生序列变成2 7 7 7 7 5 7,最长完美学生序列长度为4.
[数据范围与约定]
对于10%的数据:n<=10
对于30%的数据:n<=1000
对于100%的数据:1<=n<=100000
题意说白了就是给你n个数,可以删掉m类数,使留下来的数中连续且数字一样的序列最长。
我们可以先考虑一个区间的情况,假设一个区间有x类数,我为了保留某类数,肯定要删x-1次,因为不管保留哪类数,都要删x-1次,那我肯定保留个数最多的那一类。
但是枚举区间就要
n2
,复杂度有
n3
,明显难以承受。
再想一想,难道真的要枚举每个区间吗?这里对于区间有数的种类的限制。这和以前的头尾指针
O(n)
枚举是一样的阿!我们只要维护现有的数的种类,最大值用线段树来求,就可以完美地
O(nlogn)
解决了。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define p1 id<<1
#define p2 id<<1^1
using namespace std;
int n,m,t,sub,ans,have;
int a[100005],tree[400005],sum[100005];
struct ty
{
int v,id;
}b[100005];
bool cmp(ty x,ty y)
{
return x.v<y.v;
}
void update(int id,int l,int r,int x,int y)
{
if(l==r&&l==x)
{
tree[id]=tree[id]+y;
return;
}
int mid=(l+r)/2;
if(x<=mid) update(p1,l,mid,x,y);
else update(p2,mid+1,r,x,y);
tree[id]=max(tree[p1],tree[p2]);
}
int main()
{
cin>>n>>m;
m++;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
b[i].v=a[i];
b[i].id=i;
}
sort(b+1,b+n+1,cmp);
for(int i=1;i<=n;i++) //离散化
{
if(b[i].v!=b[i-1].v) sub++;
a[b[i].id]=sub;
}
t=1;
for(int i=1;i<=n;i++)
{
if(sum[a[i]]==0)
{
sum[a[i]]++;
have++;
}
else sum[a[i]]++;
update(1,1,sub,a[i],1);
while(have>m)
{
update(1,1,sub,a[t],-1);
sum[a[t]]--;
if(sum[a[t]]==0) have--;
t++;
}
if(have<=m) ans=max(ans,tree[1]);
}
cout<<ans;
return 0;
}