POJ 2408 Anagram Groups(Trie+思维)

 

给出多个单词,每个单词的长度没说(差评),每个单词只要所有的字母相同可以认为是同一个单词,将所有的同一个单词放入集合中,输出前 5 个集合数量最大的集合

注意:若一个集合中只有一个单词 word,不能说明 word 只出现过一次

题目不难,但是处理起来有几个坑点

  1. 题目可以用 map+set 过,但是常数较大,我们采用字典树,将每一个单词在树上的标记作为桶
  2. 将不同的字符串转化为字典序最小的,可以采用快排,这里采用基数排序
  3. 可能是 set 里面有 const 修饰的形参,所以 cmp 函数必须采用 const 的形参
const int N=1e6+5;

    int n,m;
    int i,j,k;
    char s[N];
    int t[N][26],tot=0;
    struct Node
    {
        set<string> all;
        int sz;
    }vis[N];
    int digit[26],id[N],ans=0;

int idx(char ch){ return ch-'a'; }

void insert(char *s)
{
    string str="";
    int len=strlen(s);
    for(int i=0;i<s[i];i++) digit[s[i]-'a']++; //基数排序
    for(int i=0;i<26;i++) while(digit[i]) str+=i+'a',--digit[i];
    int rt=0;
    for(int i=0;i<str[i];i++) {
        int x=idx(str[i]);
        if(!t[rt][x]) t[rt][x]=++tot;
        rt=t[rt][x];
    }
    if(vis[id[rt]].sz==0) id[rt]=++ans;
    vis[id[rt]].sz++; vis[id[rt]].all.insert((string)s);
}

bool cmp(const Node &a,const Node &b)
{
    return a.sz==b.sz?(*(a.all.begin()))<(*(b.all.begin())):a.sz>b.sz;
}

int main()
{
    //IOS;
    while(scanf("%s",s)!=EOF) insert(s);
    set<string>::iterator it;
    sort(vis+1,vis+1+ans,cmp);  
    //for(it=vis[1].all.begin();it!=vis[1].all.end();it++) dbg(*it);
    for(int i=1;i<=5;i++){
        printf("Group of size %d: ",vis[i].sz);
        for(it=vis[i].all.begin();it!=vis[i].all.end();it++) printf("%s ",(*it).c_str());
        puts(".");
    }
    PAUSE;
    return 0;
}

 

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 酷酷鲨 设计师:CSDN官方博客 返回首页