POJ - 1451 T9 字典树好题

题意：
原来的按键手机都一般是九键，九键输入英文很麻烦，例如要键入“hello”，必须按两次键4、两次键3、三次键5、三次键5，最后按三次键6。

现有一种新的输入方案名叫“T9”，只需要不重复地按键，软件就会使用内置的字典来查找最可能的与输入匹配的单词。例如，输入“hello”，只要依次按下4,3,5,5,6各一次即可。当然，这也可能是“gdjjm”一词的输入，但是因为这不是一个合理的英语单词，所以可以安全地忽略它。通过排除所有其他不可能的解决方案，并且只考虑适当的英语单词，这种方法可以大大加快短信的写作速度。当然，如果这个词不在软件内置的字典中（比如名字），那么它必须再次使用多次按键的方式输入。

现在给你“T9”的字典，包含 w(0≤w≤1e3) 个互不相同的字符串（串长不超过 100，已经按字典序升序排序），以及他们的出现概率 p。又给出 m 次打字操作，每次输入包含不超过 100 个数字的 2∼9 数字串，代表依次按键，最后跟一个 1 代表结束本次打字。

对“字符组合”的概率，定义为所有以该字符组合为前缀的单词的出现概率之和。例如，如果字典包含三个单词“hell”,“hello”和“hellfire”，则字符组合“hell”的概率是这些词的概率之和。如果某些组合具有相同的概率，则您的程序是选字典序最小的。
思路：
因为已经排好序了，我们先遍历一遍计算出每个前缀的概率和，把每个前缀的概率和存在公共前缀的字典序最大的那个字符串里，
然后插入到字典树中
我们用每个字母对应的数字来建树，但是由于每个数字对应3-4的字母，所以我们在该节点中存：到目前为止 概率最大的串，建树的过程中概率高的覆盖低的就可以了。
查询的时候查到哪输出哪

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#include<set>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=1e5+5;
int trie[maxn][15],idx;
char cun[1005][105];
char mp[]="22233344455566677778889999";
int nowval[maxn];
int val[1005][105];
void insert(char s[],int x)
{
	int u=0,k=0;
	char temp[1005];
	for(int i=0;s[i];i++)
	{
		int v=mp[s[i]-'a']-'0';
		temp[k++]=s[i];
		if(!trie[u][v])
			trie[u][v]=++idx;
		u=trie[u][v];
		if(nowval[u]<val[x][i])
		{
			temp[k]='\0';
			nowval[u]=val[x][i];
			strcpy(cun[u],temp);
		}
	}
}
void query(char s[])
{
	int u=0,i;
	for(i=0;s[i];i++)
	{
		int v=s[i]-'0';
		if(!trie[u][v])
		{
			i++;
			break;
		}
		u=trie[u][v];
		printf("%s\n",cun[u]);
	}
	for(;s[i];i++)
		printf("MANUALLY\n");
	printf("\n");
}
char str[1005][105];
void init()
{
	idx=0;
	memset(trie,0,sizeof trie);
	memset(nowval,0,sizeof nowval);
	memset(val,0,sizeof val);
}
int main()
{
	int t,ok=1;
	scanf("%d",&t);
	while(t--)
	{
		int n;
		init();
		scanf("%d",&n);
		for(int i=0;i<n;i++)
		{
			int p;
			scanf("%s%d",str[i],&p);
			for(int j=0;str[i][j];j++)
				val[i][j]=p;
		}
		for(int i=1;i<n;i++)
		{
			for(int j=0;str[i][j]&&str[i-1][j];j++)
			{
				if(str[i][j]==str[i-1][j])
				{
					val[i][j]+=val[i-1][j];
					val[i-1][j]=0;
				}
				else break;
			}
		}
		for(int i=0;i<n;i++)
			insert(str[i],i);
		scanf("%d",&n);
		printf("Scenario #%d:\n",ok++);
		for(int i=0;i<n;i++)
		{
			char s[100];
			scanf("%s",s);
			query(s);
		}
		printf("\n");
	}
	return 0;
}
/*
1
2
give 13
integer 6
1
4681
*/