2021HDU多校第五场1004 Another String-双指针

https://acm.hdu.edu.cn/showproblem.php?pid=7015

 题意：对于每次以i来分割字符串，计算两个分割串的子串的k-匹配的数量。

思路：当我们计算到ans[i]是，我们可以当成从ans[i-1]加上【1-i】字符串和【i+1，n】的k-匹配数量再减去【1-（i-1）】和【i-n】的k-匹配的数量。

sum1[i]：【1-（i-1）】和【i-n】的k-匹配的数量

sum2[i]:【1-i】和【i+1，n】的k-匹配数量

那我们怎么算这个呢，我们设dp1[i][j]代表从i开始和从j开始k-匹配的数量（方向是从1—>n）

然后sum1[i] 就是dp1[0-(i-1)][i]的和，我们求一下前缀和就好了

计算sum2[i]同理，设dp2[i][j]表示反方向从i开始和和从j开始 k-匹配的数量（反向是从n—>1）

然后sum2[i] 就是dp2[(i+1)-n][i]的和

关键点在于我们如何求出dp1[i][j]，dp2[i][j]

我们可以枚举d表示他们的距离，每次求出dp1[i][i+d]，和dp2[i][i-d]

然后使用双指针优化，dp1[i][i+d]能匹配到 j，那么dp1[i+1][i+1+d]比然也能匹配到 j，那我们下一个可以直接从开始j往后移。

#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <queue>
#include <cstdlib>
#include <stack>
#include <vector>
#include <set>
#include <map>
#include <bitset>
#include <complex>
#define INF 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
#define FILL(a,b) (memset(a,b,sizeof(a)))
#define lson rt<<1
#define rson rt<<1|1
#define lowbit(a) ((a)&-(a))
#define ios std::ios::sync_with_stdio(false);std::cin.tie(0);std::cout.tie(0);
#define fi first
#define sc second
#define pb push_back
#define all(x) (x).begin(),(x).end()
using namespace std;
typedef long long  ll;
typedef unsigned long long  ull;
typedef pair<int,int> pii;
const ll mod=1e9+7;
const ll N =6e6+10;
const double eps = 1e-6;
const double PI=acos(-1);
ll gcd(ll a,ll b){return !b?a:gcd(b,a%b);}
ll re(){ll x;scanf("%lld",&x);return x;}
int dx[8]= {1,0,-1,0,1,1,-1,-1}, dy[8] = {0,1,0,-1,1,-1,1,-1};
int n,k;
int dp1[3100][3100],dp2[3100][3100];
ll sum1[3100],sum2[3100];
char s[3100];
void solve(){
    cin>>n>>k;
    scanf("%s",s+1);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            dp1[i][j]=dp2[i][j]=0;
        }
        sum1[i]=sum2[i]=0;
    }
    for(int i=1;i<n;i++){
        if(s[i]==s[i+1]||k>0) dp1[i][i+1]=1;
        if(s[i]==s[i+1]||k>0) dp2[i+1][i]=1;
    }
    for(int d=2;d<n;d++){
        int j=-1;
        int x=0;
        for(int i=1;i+d<=n;i++){
            while(x<=k&&i+d+j<=n&&j<d){
                j++;
                if(i+d+j>n) break;
                if(s[i+j]!=s[i+d+j]) x++;
            }
            dp1[i][i+d]=j;
            if(s[i]!=s[i+d]) x--;
            j--;
        }
    }
    for(int d=2;d<n;d++){
        int j=-1;
        int x=0;
        for(int i=n;i-d>=1;i--){
            while(x<=k&&i-d-j>=1&&j<d){
                j++;
                if(i-d-j<1) break;
                if(s[i-j]!=s[i-d-j]) x++;
            }
            dp2[i][i-d]=j;
            if(s[i]!=s[i-d]) x--;
            j--;
        }
    }
    for(int i=1;i<n;i++){
        for(int j=i+1;j<=n;j++){
            sum2[i]+=dp2[j][i];
        }
    }
    for(int i=2;i<=n;i++){
        for(int j=1;j<i;j++){
            sum1[i]+=dp1[j][i];
        }
    }
    ll ans=0;
    for(int i=1;i<n;i++){
        ans+=(sum2[i]-sum1[i]);
        cout<<ans<<endl;
    }
}
int main()
{
  //  ios
    cin>>t;
    while(t--) solve();
    return 0;
}