hdu_1298 字典树

来源：互联网发布：java 静态构造函数编辑：程序博客网时间：2024/05/29 02:58

这道字典树稍微难一点~主要是注意重载运算符的时候如果他们的概率相同就比较他们的字典序，如果字典序小则相当于概率越大，所以重载的时候要用大于号就可以~~

其他的话就是注意格式，输出的空格的数目，至于怎样计算每个字符串的概率只要稍微观察一下就能得出一个字符串的概率等于所有以它为前缀的字符串的概率之和~~是吧，它也是它本身的前缀~~~ 所以编写的时候就很好编了，插入的时候只要经过一个字符串的话，这个字符串就加上插入字符串的概率即可。

#include<iostream>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;
int T,w,np,n,apm;
const int ax='a';
string a,b,c;
struct pp
{
string s;
int num;
bool operator < (const pp &cmp) const
{
if(num!=cmp.num)
{
return num<cmp.num;
}
else
{
return s>cmp.s;
}
}
bool operator == (const pp& cmp) const
{
return num==cmp.num;
}
bool operator <= (const pp & cmp) const
{
return num<=cmp.num;
}
}px;
vector<pp>vp;
struct tree
{
int num;
tree *p[26];
tree()
{
num=0;
for(int i=0;i<26;i++)
{
p[i]=NULL;
}
}
}*head,*now;
void insert(string x,int n)
{
for(int i=0;i<x.length();i++)
{
if(i==0)
{
if(head->p[x[i]-ax]==NULL)
{
head->p[x[i]-ax]=new tree;
now=head->p[x[i]-ax];
now->num+=n;
}
else
{
now=head->p[x[i]-ax];
now->num+=n;
}
}
else
{
if(now->p[x[i]-ax]==NULL)
{
now->p[x[i]-ax]=new tree;
now=now->p[x[i]-ax];
now->num+=n;
}
else
{
now=now->p[x[i]-ax];
now->num+=n;
}
}
}
}
void fuck(string x="",tree* temp=head,int in=0)
{
if(in==n)
{
px.num=temp->num;
px.s=x;
vp.push_back(px);
}
else
{
if(b[in]=='2')
{
for(int i=0;i<=2;i++)
{
if(temp->p[i]!=NULL)
{
fuck(x+char(ax+i),temp->p[i],in+1);
}
else
{
continue;
}
}
}
else if(b[in]=='3')
{
for(int i=3;i<=5;i++)
{
if(temp->p[i]!=NULL)
{
fuck(x+char(ax+i),temp->p[i],in+1);
}
else
{
continue;
}
}
}
else if(b[in]=='4')
{
for(int i=6;i<=8;i++)
{
if(temp->p[i]!=NULL)
{
fuck(x+char(ax+i),temp->p[i],in+1);
}
else
{
continue;
}
}

}
else if(b[in]=='5')
{
for(int i=9;i<=11;i++)
{
if(temp->p[i]!=NULL)
{
fuck(x+char(ax+i),temp->p[i],in+1);
}
else
{
continue;
}
}

}
else if(b[in]=='6')
{
for(int i=12;i<=14;i++)
{
if(temp->p[i]!=NULL)
{
fuck(x+char(ax+i),temp->p[i],in+1);
}
else
{
continue;
}
}
}
else if(b[in]=='7')
{
for(int i=15;i<=18;i++)
{
if(temp->p[i]!=NULL)
{
fuck(x+char(ax+i),temp->p[i],in+1);
}
else
{
continue;
}
}
}
else if(b[in]=='8')
{
for(int i=19;i<=21;i++)
{
if(temp->p[i]!=NULL)
{
fuck(x+char(ax+i),temp->p[i],in+1);
}
else
{
continue;
}
}
}
else if(b[in]=='9')
{
for(int i=22;i<=25;i++)
{
if(temp->p[i]!=NULL)
{
fuck(x+char(ax+i),temp->p[i],in+1);
}
else
{
continue;
}
}
}
}
}
int main()
{
cin>>T;
for(int tt=1;tt<=T;tt++)
{
head=new tree;
cin>>w;
for(int i=1;i<=w;i++)
{
cin>>a;
cin>>np;
insert(a,np);
}
cin>>w;
cout<<"Scenario #"<<tt<<":\n";
for(int i=1;i<=w;i++)
{
cin>>b;
for(n=1;n<b.length();n++)
{
vp.clear();
fuck();
if(vp.empty())
{
cout<<"MANUALLY"<<endl;
}
else
{
sort(vp.begin(),vp.end());
cout<<vp.back().s<<endl;
}
}
cout<<endl;
}
cout<<endl;
}
return 0;
}