P3499【2015多校联训6】密码（链表模拟）

问题描述

假发通过了不懈的努力，得到了将军家门锁的密码（一串小写英文字母）。但是假发被 十四和猩猩他们盯上了，所以假发需要把密码传递出去。因为假发不想十四他们发现几松门 前贴的小纸条就是将军家的密码，所以他加密了密码（新八：听起来有点诡异）。加密方法 如下：随机地，在密码中任意位置插入随机长度的小写字符串。 不过，假发相信银桑和他那么多年小学同学，一定能猜中密码是什么的（新八：银桑什 么时候成攮夷志士了！！！）。可是，写完了小纸条之后，假发觉得有点长，就想截去头和 尾各一段（可以为空），让剩下的中间那一段依然包含真～密码。想着想着，假发就想知道 有多少种可行方案。结果在沉迷于稿纸之际，假发被投进了狱门岛（新八：……）。于是， 就由你计算了。

输入格式

两行非空字符串，纯小写英文字母，第一行是加密后的密码，第二行是原密码。
第一行长度不超过 300000，第二行不超过 200。

输出格式

一行，有多少种方案。注意：不剪也是一种方案。

样例输入 1

abcabcabc
cba

样例输出 1

9

样例输入 2

abcabcabac
cba

样例输出 2

18

题解

用前向星记录下当前位置到下一个字母的位置

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<queue>
#include<vector>
using namespace std;
#define maxn 3000005
#define int long long
int n,m;
int Ne[maxn][26];
int la[26];
int ans;
string a,key;
main()
{
    int i,j;
    cin>>a>>key;
    n=a.length();m=key.length();
    a=' '+a;key=' '+key;
    for(i=0;i<=25;i++) la[i]=n+1;
    for(i=n;i>=1;i--){
        for(j=0;j<26;j++) Ne[i][j]=la[j];
        la[a[i]-'a']=i;
    }
    int la=0;
    int cnt,x;
    for(i=1;i<=n;i++){
        if(a[i]==key[1]){
            x=i,cnt=1,j=1;
            while(x<=n&&++cnt<=m){

                x=Ne[x][key[cnt]-'a'];

            }
            if(x<=n){
//              cout<<i<<" "<<x<<endl;
                ans+=(i-la)*(n-x+1);
                la=i;
            }
        }
    }
    cout<<ans;
}