听说hash做法很简单。。。不管了,我写的是数据结构。。。
以下不要计较开闭区间等等。
考虑从左往右枚举右切口
r
,假设左切口
对于满足第一种情况的颜色,记
prec
为其
r
在之前最后一次出现的位置,那么要满足
对于满足第二种情况的颜色,记
firstc
和
lastc
为其第一次和最后一次出现的位置,那么首先要满足
r>=lastc
,然后要在优先队列中删除
c
出现的那些位置,然后
于是问题就是维护一个带删除操作的优先队列,并且会不断给一些区间标记,并且询问一个区间
[max{prec},r]
中有多少个未标记的点,用线段树即可。
对于第二问,我们考虑尽量让切的长度接近
n/2
,于是就在
r−n/2
左右找到第一个未被标记得点即可,也是线段树的经典操作。
代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;
const int maxn=1000010;
int n,c[maxn],pre[maxn],lst[maxn],k;
vector<int> a[maxn];
struct heap
{
priority_queue<int> Q,D;
void push(int x) {Q.push(x);}
void erase(int x) {D.push(x);}
int top()
{
while(!D.empty()&&Q.top()==D.top()) Q.pop(),D.pop();
return Q.empty()?0:Q.top();
}
}H;
struct tree
{
int l,r,sum;
bool b,cvr;
tree *ls,*rs;
tree(){ls=rs=NULL;cvr=0;}
void cal(){sum=0;b=0;}
void update(){b=ls->b||rs->b; sum=ls->sum+rs->sum;}
void pushdown()
{
if(cvr) ls->cvr=rs->cvr=1,ls->cal(),rs->cal(),cvr=0;
}
void build(int lx,int rx)
{
l=lx;r=rx;b=1;sum=r-l+1;
if(l==r) return;
int mid=(l+r)>>1;
(ls=new tree)->build(lx,mid);
(rs=new tree)->build(mid+1,rx);
}
void del(int lx,int rx)
{
if(lx>rx) return;
if(lx==l&&rx==r) {cvr=1;cal();return;}
pushdown();
int mid=(l+r)>>1;
if(rx<=mid) ls->del(lx,rx);
else if(lx>mid) rs->del(lx,rx);
else ls->del(lx,mid),rs->del(mid+1,rx);
update();
}
int find(int pl,bool f)
{
if(!b) return -1;
if(l==r) return l;
pushdown();
int mid=(l+r)>>1,tmp=-1;
if(f)
{
if(pl>mid) tmp=rs->find(pl,f);
if(tmp==-1) tmp=ls->find(pl,f);
}
else
{
if(pl<=mid) tmp=ls->find(pl,f);
if(tmp==-1) tmp=rs->find(pl,f);
}
return tmp;
}
int qry(int lx,int rx)
{
if(lx>rx) return 0;
if(l==lx&&r==rx) return sum;
pushdown();
int mid=(l+r)>>1;
if(rx<=mid) return ls->qry(lx,rx);
else if(lx>mid) return rs->qry(lx,rx);
else return ls->qry(lx,mid)+rs->qry(mid+1,rx);
}
}*xtr;
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
scanf("%d",&c[i]);
(xtr=new tree)->build(1,n);
for(int i=n;i;i--)
a[c[i]].push_back(i);
long long ans1=0;int ans2=n;
for(int i=1;i<n;i++)
{
H.push(i);
if(a[c[i]][0]==i)
{
for(int j=0;j<a[c[i]].size();j++)
H.erase(a[c[i]][j]);
xtr->del(*(a[c[i]].end()-1)+1,a[c[i]][0]);
}
int lft=H.top()+1,fl=xtr->find(max(lft,i-n/2+1),1),fr=xtr->find(max(lft,i-n/2+1),0);
ans1+=xtr->qry(lft,i);
if(fl>=lft) ans2=min(ans2,abs(n-2*(i-fl+1)));
if(fr>=lft) ans2=min(ans2,abs(n-2*(i-fr+1)));
}
printf("%lld %d",ans1,ans2);
return 0;
}