【jzoj5093】【GDSOI2017第四轮模拟day3】【字符串匹配】【哈希】

题目大意

对于一个字符集大小为C的字符串P,我们可以将任意两种字符在P中的位置进行互换,例如P=abcba,我们交换a,b就变为bacab,交换a,d就变为dbcbd,交换可以进行任意次。若交换后P变为了字符串Q,则我们称Q与P是匹配的。

现在给定两个字符集大小为C的字符串S,T,请你求出S中有多少个连续子串与T是匹配的。

解题思路

每个位的值为与前一个相同颜色的位置差,对这个哈希即可判断是否同构,特殊处理一下关联关系即可支持加入和删除。

code

#include
#include
#define LL long long
#define ULL unsigned long long
#define fo(i,j,k) for(int i=j;i<=k;i++)
using namespace std;
int const mn=1e6+9,inf=1e9+7;
int t,c,n,m,ans[mn],a[mn],b[mn],pre[mn],next[mn];
ULL p=1e9+7,bit[mn];
int main(){
    //freopen("string.in","r",stdin);
    //freopen("string.out","w",stdout);
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    scanf("%d%d",&t,&c);
    bit[0]=1;fo(i,1,1000000)bit[i]=bit[i-1]*p;
    fo(cas,1,t){
        scanf("%d%d",&n,&m);
        fo(i,1,n)scanf("%d",&a[i]);
        fo(i,1,m)scanf("%d",&b[i]);
        fo(i,1,c)pre[i]=0;
        ULL hashb=0;
        fo(i,1,m){
            int tmp=(pre[b[i]])?i-pre[b[i]]:0;
            hashb=hashb*p+tmp;
            pre[b[i]]=i;
        }
        fo(i,1,c)pre[i]=0;
        fo(i,1,n)next[i]=0;
        ULL hasha=0;
        fo(i,1,m){
            int tmp=(pre[a[i]])?i-pre[a[i]]:0;
            hasha=hasha*p+tmp;
            next[pre[a[i]]]=i;
            pre[a[i]]=i;
        }
        ans[0]=0;
        if(hasha==hashb)ans[++ans[0]]=1;
        fo(i,m+1,n){
            int ii=i-m,tmp=(pre[a[i]]>ii)?i-pre[a[i]]:0,tmp2=(next[ii])?next[ii]-ii:0;
            hasha=(hasha-tmp2*bit[i-1-next[ii]])*p+tmp;
            if(hasha==hashb)ans[++ans[0]]=ii+1;
            next[pre[a[i]]]=i;
            pre[a[i]]=i;
        }
        printf("%d\n",ans[0]);
        fo(i,1,ans[0])printf("%d ",ans[i]);
        printf("\n");
    }
    return 0;
}

你可能感兴趣的:(jzoj,字符串)