hdu3689(kmp+概率dp)

链接：点击打开链接

题意：给出n个字母出现的频率和一个字符串s，求长度为m的字符串含有s的概率

代码：

#include <set>
#include <map>
#include <queue>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
char s[1005];
int len,nex[1005];
double dp[1005][1005];
map<char,double> mp;
map<char,double>::iterator ite;
void getnext(){
    int i,j;
    i=0,j=-1;
    nex[i]=j;
    while(i<len){
        if(j==-1||s[i+1]==s[j+1])
        nex[++i]=++j;
        else
        j=nex[j];
    }                                           //nex数组也从下标1开始
}
int main(){
    char c,fir;
    int n,m,i,j,tt;
    double pp,ans,sec;
    while(scanf("%d%d",&n,&m)!=EOF&&(n||m)){
        mp.clear();
        for(i=1;i<=n;i++){
            cin>>c>>pp;
            mp[c]=pp;
        }
        scanf("%s",s+1);
        len=strlen(s+1);
        getnext();
        for(i=0;i<=m;i++)                       //dp[i][j]表示长度是i，匹配到子串的j的概率
        for(j=0;j<=len;j++)                     //所以只要求出nex数组直接转移即可，但要注意
        dp[i][j]=0;                             //当第i位匹配到子串最后一位，不要继续转移，否则
        dp[0][0]=1;                             //后面的概率可能变成条件概率
        for(i=0;i<m;i++){
            for(j=0;j<len;j++){                 //就一个子串，没必要直接上自动机...
                if(dp[i][j]){
                    for(ite=mp.begin();ite!=mp.end();ite++){
                        fir=ite->first;
                        sec=ite->second;
                        if(fir==s[j+1])
                        dp[i+1][j+1]+=(dp[i][j]*sec);
                        else{
                            tt=j;
                            while(tt&&s[tt+1]!=fir)
                            tt=nex[tt];
                            if(s[tt+1]==fir)
                            tt++;
                            dp[i+1][tt]+=(dp[i][j]*sec);
                        }
                    }
                }
            }
        }
        ans=0;
        for(i=1;i<=m;i++)
        ans+=dp[i][len];
        printf("%.2lf%%\n",ans*100);
    }
    return 0;
}