USACO Name That Number (dfs+字典树)

题目大意：2到9每个数字对应3字母（对应关系题中已给出），给定一个长度不超过12的数字序列，则可以对应到多个字符串，现给定一个字典（含很多字符串的文件），打印数字序列对应的且在字典中出现过的字符串。

分析：每个数字对应3个字母，最坏情况下有12个数字，可对应3¹²=531441个字符串，而且题中字典中的字符串是按字典序给出的，所以可以用二分查找，所以应该可以用暴力+二分过。另一个办法就是建字典树，然后dfs时边搜遍判断，这样效率应该有所提升。我是用字典树过的。

需要注意的是当字典中不存在符合要求的字符串时，要输出“NONE”

View Code

/*

ID: lijian42

LANG: C++

TASK: namenum

*/

#include <stdio.h>

#include <string.h>

#define N 5001

#define LEN 15

int node;

int next[N*LEN][26];

bool isend[N*LEN];

int n;

char num[LEN];

char name[LEN];

char table[10][3];

bool success;

void add(int cur,int k)

{

    memset(next[node],0,sizeof(next[node]));

    isend[node]=false;

    next[cur][k]=node++;

}

void build()

{

    FILE *fp=fopen("dict.txt","r");



    int i,k,cur;

    char s[LEN];

    node=1;

    memset(next[0],0,sizeof(next[0]));

    while(~fscanf(fp,"%s",s))

    {

        cur=0;

        for(i=0;s[i];i++)

        {

            k=s[i]-'A';

            if(next[cur][k]==0) add(cur,k);

            cur=next[cur][k];

        }

        isend[cur]=true;

    }

}

void init()

{

    int i,j;

    char c='A';

    for(i=2;i<10;i++)

    {

        for(j=0;j<3;j++)

        {

            table[i][j]=c;

            c++;

            if(c=='Q')  c++;

        }

    }

    build();

}

void dfs(int dep,int cur)

{

    if(dep==n)

    {

        name[dep]=0;

        if(isend[cur]) puts(name),success=true;

        return;

    }

    int i=num[dep]-'0',j,k;

    for(j=0;j<3;j++)

    {

        k=table[i][j]-'A';

        if(next[cur][k]==0) continue;

        name[dep]=k+'A';

        dfs(dep+1,next[cur][k]);

    }

}

int main()

{

    freopen("namenum.in","r",stdin);

    freopen("namenum.out","w",stdout);

    init();

    while(~scanf("%s",num))

    {

        n=strlen(num);

        success=false;

        dfs(0,0);

        if(success==false)  puts("NONE");

    }

    return 0;

}