https://ac.nowcoder.com/acm/contest/11169/D
思路:
作者:hx073269
链接:https://ac.nowcoder.com/discuss/625099?type=101&channel=-1&source_id=0
来源:牛客网
考点:动态规划+字符串hash
设DP[i]为前i个字母组成的字符串最少划分为DP[i]段,那么最终答案就是DP[n],转移可以分为以下两种:
①因为长度小于D的字符串均为D型回文串,因此有DP[i]=min(DP[i],min(DP[i-1],DP[i-2],...,DP[i-D+1])+1)。这里可以使用一个单调队列维护定长区间最小值。
②定义一个位置i是“好的”,当且仅当[i-D+1,i]是回文串。若[i-D+1,i]区间是回文串,则令mi=min(mi,DP[i-D]);若不是,就令mi=INF。因为“好的”位置是连续的,所以我们直接让mi随着i的增大而更新,同时每次令DP[i]=min(DP[i],mi+1)即可。这里可以使用字符串hash预处理后快速判断回文串。
上面的思路比较自然,实际上可以发现DP数组是单调不减的,因此只需要求最小的l,使得[l,i]为D型回文串,那么令DP[i]=DP[l-1]+1即可。故我们不需要使用单调队列也可完成转移。
时间复杂度:O(n)。
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e7+100;
typedef long long LL;
const LL base=131;
const LL mod=998244353;
inline LL read(){LL x=0,f=1;char ch=getchar(); while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL h[maxn],p1[maxn],p2[maxn];
char str[maxn];
LL dp[maxn];
LL Q[maxn],t=0,f=0;///t--队头,f---队尾
LL check(LL p[],LL l,LL r){
return (p[r]%mod-p[l-1]*h[r-l+1]%mod+mod)%mod;
}
int main(void)
{
cin.tie(0);std::ios::sync_with_stdio(false);
LL n,D;cin>>n>>D;
cin>>(str+1);
if(D>n){
cout<<"1"<<"\n";return 0;
}
h[0]=1;
LL mi=1e9;
for(int i=1;i<maxn;i++) h[i]=h[i-1]*base%mod;
for(int i=1;i<=n;i++) p1[i]=(p1[i-1]*base%mod+(str[i]-'a'+1))%mod,p2[i]=(p2[i-1]*base%mod+(str[n-i+1]-'a'+1))%mod;
for(int i=1;i<=n;i++){
dp[i]=1e9;
if(i<D){
dp[i]=1;
}
else{
if(t>f) dp[i]=min(dp[i],dp[Q[f]]+1);
if(check(p1,i-D+1,i)==check(p2,n-i+1,n-i+D)) mi=min(mi,dp[i-D]);
else mi=1e9;
dp[i]=min(dp[i],mi+1);
}
while(t>f&&dp[Q[t-1]]>=dp[i]) t--;
Q[t++]=i;
while(t>f&&i-Q[f]+1>=D) f++;
}
cout<<dp[n]<<"\n";
return 0;
}