ZOJ 1387 Decoding Morse Sequences

水动规= =

#include <cstdio>
#include <cstring>
#include <cstdlib>

const int MAXN = 10010;
const char dic[26][10] =
{
    ".-", "-...", "-.-.", "-..",
    ".", "..-.", "--.", "....",
    "..", ".---", "-.-", ".-..",
    "--", "-.", "---", ".--.",
    "--.-", ".-.", "...", "-",
    "..-", "...-", ".--", "-..-",
    "-.--", "--.."
};

char str[MAXN];
char morse[MAXN][110];
int diclen[26];
int dp[MAXN];
int N;

void init()
{
    for ( int i = 0; i < 26; ++i )
        diclen[i] = strlen( dic[i] );
    return;
}

void DP()
{
    int len = strlen( &str[1] );
    memset( dp, 0, sizeof(dp) );
    dp[0] = 1;
    for ( int i = 1; i <= len; ++i )
        for ( int j = 0; j < N; ++j )
        {
            bool ok = true;
            int k;
            //printf( "morse[%d]=%s\n", j, morse[j] );
            for ( k = 0; morse[j][k]; ++k )
                if ( str[i + k] != morse[j][k] )
                {
                    ok = false;
                    break;
                }
            if ( ok ) dp[ i + k - 1 ] += dp[i - 1];
        }
    //for ( int i = 0; i <= len; ++i ) printf( "dp[%d] = %d\n", i, dp[i] );

    printf( "%d\n", dp[len] );
    return;
}

int main()
{
    init();

    int T;
    scanf( "%d", &T );
    while ( T-- )
    {
        scanf( "%s", &str[1] );
        scanf( "%d", &N );
        for ( int i = 0; i < N; ++i )
        {
            char temp[24];
            scanf( "%s", temp );
            int lenn = 0;
            for ( int j = 0; temp[j]; ++j )
            {
                strcpy( &morse[i][lenn], dic[ temp[j] - 'A' ] );
                lenn += diclen[ temp[j] - 'A' ];
            }
            morse[i][lenn] = '\0';
            //puts( morse[i] );
        }
        DP();
    }
    return 0;
}