poj 2945 Find the Clones

https://vjudge.net/problem/POJ-2945

题意：

给出n个长度相同的DNA序列，如果一个DNA序列出现过两次，那么就有说明它被复制了一次。问被复制0次，1次，2次……n-1次的DNA序列分别有多少个。

思路：

可以利用字典树的方法做，用map目测会超时。因为一个字符串是它自己本身的假前缀，所以把每一个字符串的前缀数量-1，就是它被复制的数目。这题学到了一个技巧，统计的时候，已经统计过的要跳开，一开始用map判断超时了，之后想到用sort对char类型的字符串进行排序，但是sort是不能对char的字符串进行排序。但是，我们有结构体啊，把字符串封装在一个结构体中，自己写一个cmp函数就完美的解决了这个问题。这题还是字典树模板用上啦。一定记住最后要释放内存啊。QAQ

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

int a[20005];

const int maxn = 26;

struct str
{
    char s[25];
}b[20005];

bool cmp(str aa,str bb)
{
    return strcmp(aa.s,bb.s) < 0;
}

struct trie
{
    trie *next[maxn];

    int flag;

    trie()
    {
        flag = 1;
        memset(next,NULL,sizeof(next));
    }
}*root;

void insert(char *str)
{
    int len = strlen(str);
    trie *p = root,*q;

    for (int i = 0;i < len;i++)
    {
        int id = str[i] - 'A';

        if (p -> next[id] == NULL)
        {
            q = new trie();
            p -> next[id] = q;
            p = p -> next[id];
        }
        else
        {
            p = p->next[id];

            ++(p->flag);
        }
    }
}

int query(char *str)
{
    int len = strlen(str);
    trie *p = root;

    for (int i = 0;i < len;i++)
    {
        int id = str[i] - 'A';

        p = p -> next[id];

        if (p == NULL) return 0;
    }

    return p->flag;
}

void Free(trie* t)
{
    if (t == NULL) return;

    for (int i = 0;i < maxn;i++)
    {
        if (t -> next[i]) Free(t->next[i]);
    }

    delete(t);
}



int main()
{
    int n,m;

    while (scanf("%d%d",&n,&m) != EOF)
    {
        if (m == 0 && n == 0) break;

        memset(a,0,sizeof(a));

        root = new trie();

        for (int i = 0;i < n;i++)
        {
            scanf("%s",b[i].s);
            insert(b[i].s);
        }

        sort(b,b+n,cmp);

        //for (int i = 0;i < n;i++)
            //printf("%s
",b[i].s);

        int rs = query(b[0].s);

        char ss[25];strcpy(ss,b[0].s);

        a[rs-1]++;

        for (int i = 1;i < n;i++)
        {
            if (strcmp(ss,b[i].s) == 0) continue;

            strcpy(ss,b[i].s);

            rs = query(b[i].s);

            a[rs-1]++;
        }


        for (int i = 0;i < n;i++)
        {
            printf("%d
",a[i]);
        }

        Free(root);
    }

    return 0;
}