Description
给出一个字符串,对任一前缀,问最少可以将其划分成几个合法串,定义合法串为其恰含k个不同的字符
Input
第一行首先输入一整数k表示对合法串的要求,之后一个全由小写字母组成的字符串s(1<=k<=26,1<=|s|<=2e5)
Output
输出|s|个数,第i个数表示前缀[1,i]最少可以分成几个合法串,如果无解输出-1
Sample Input
2
abacaba
Sample Output
-1 1 1 2 2 3 3
Solution
对每个i,需要求出区间[l[i],r[i]]使得对这个区间的任意j,[j,i]区间恰有k个不同字母,求l[i]和r[i]可以通过两边尺取得到,以dp[i]表示前缀[1,i]至少可以被划分成多少个合法串,那么dp[i]=min(dp[j-1])+1,其中j介于[l[i],r[i]],所以问题变成维护一个区间最小值以及单点修改,用线段树维护dp序列即可
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 222222
int n,k,num[26],a[maxn],b[maxn],Min[maxn<<2];
char s[maxn];
#define ls (t<<1)
#define rs ((t<<1)|1)
void build(int l,int r,int t)
{
Min[t]=INF;
if(l==r)return ;
int mid=(l+r)>>1;
build(l,mid,ls),build(mid+1,r,rs);
}
void push_up(int t)
{
Min[t]=min(Min[ls],Min[rs]);
}
void update(int x,int l,int r,int t,int v)
{
if(l==r)Min[t]=v;
else
{
int mid=(l+r)>>1;
if(x<=mid)update(x,l,mid,ls,v);
else update(x,mid+1,r,rs,v);
push_up(t);
}
}
int query(int L,int R,int l,int r,int t)
{
if(L<=l&&r<=R)return Min[t];
int ans=INF,mid=(l+r)>>1;
if(L<=mid)ans=min(ans,query(L,R,l,mid,ls));
if(R>mid)ans=min(ans,query(L,R,mid+1,r,rs));
return ans;
}
int main()
{
int Case=1;
while(~scanf("%d",&k))
{
scanf("%s",s+1);
n=strlen(s+1);
memset(num,0,sizeof(num));
memset(a,-1,sizeof(a));
memset(b,-1,sizeof(b));
int res=0,l=n;
for(int r=n;r>=1;r--)
{
while(res<=k&&l>0)
{
if(num[s[l]-'a']==0)res++;
num[s[l]-'a']++;
l--;
}
if(res==k+1)a[r]=l+2;
else if(res==k&&l==0)a[r]=l+1;
if(num[s[r]-'a']==1)res--;
num[s[r]-'a']--;
}
memset(num,0,sizeof(num));
l=n;
for(int r=n;r>=1;r--)
{
while(res<k&&l>0)
{
if(num[s[l]-'a']==0)res++;
num[s[l]-'a']++;
l--;
}
if(res==k)b[r]=l+1;
if(num[s[r]-'a']==1)res--;
num[s[r]-'a']--;
}
build(0,n,1);
for(int i=1;i<=n;i++)
{
int ans=-1;
if(a[i]==1)
{
ans=1;
update(i,0,n,1,1);
}
else if(a[i]!=-1)
{
ans=query(a[i]-1,b[i]-1,0,n,1);
if(ans!=INF)
{
ans++;
update(i,0,n,1,ans);
}
else ans=-1;
}
printf("%d%c",ans,i==n?'\n':' ');
}
}
return 0;
}