POJ2945链式字典树解决方案

最新推荐文章于 2020-08-04 10:00:54 发布

KhazixW2

最新推荐文章于 2020-08-04 10:00:54 发布

阅读量161

点赞数

分类专栏： POJ 字典树文章标签： POJ2945

本文链接：https://blog.csdn.net/qq_41032846/article/details/95309322

版权

POJ 同时被 2 个专栏收录

4 篇文章 0 订阅

订阅专栏

字典树

1 篇文章 0 订阅

订阅专栏

POJ2945链式字典树解决方案

首先，我们的思路是用链式的字典树结构，解决POJ2945这道题

题意是，统计所有的字符串出现的次数，并依次输出各个次数的数量

例如:

input

9 6
AAAAAA
ACACAC
GTTTTG
ACACAC
GTTTTG
ACACAC
ACACAC
TCCCCC
TCCCCC

out

遇到的一些小麻烦 : 之前提交poj爆内存了,就试了一下这个代码，结果还是爆内存，后来发现，我这里不能设置位 Node *next[26],不然铁定爆内存，于是我认真审题，发现这里设置为4就好啦因为只有4个字符需要入树，至于怎么判断下标，我们用一个fun()函数来解决就好了。
具体的思路就是 : 先用find()判断，是否在树里有该个DNA,如果有让，count[i]–,i++,;count[i]++, 如果没有，用T_insert插入,并给count[1]置为1；
最后输出for(i::n)printf("%d\n",count[i]);就可以了。

#include <iostream>
#include <cstdio>
#pragma comment(linker, "/STACK:1024000000,1024000000")　　　　
#include <cstring>
using namespace std;

struct Node
{
    Node *next[4];　　　　　
    char val;　　　　　　　　　　
    int count;
    Node(char val)
    {
        this->val =val;
        count=0;
        memset(next,NULL,sizeof(next));
    }
};

Node *root =new Node('\0');
int *count;
//int *count;
int   fun(char s)
{
    switch(s)
    {
        case 'A':return 0;break;
        case 'C':return 1;break;
        case 'G':return 2;break;
        case 'T':return 3;break;
    }
}
bool T_Find(char *s)
{
    int len=strlen(s);
    Node *p=root;
    for(int i=0;i<len;i++)
    {

        int pos=fun(s[i]);
        if(p->next[pos]==NULL)
            return false;
        p=p->next[pos];
    }
    count[p->count]--;      //如果 程序运行到最后，出现过两次了，那么 p->count 出现过一次的 count[p->count] -- ,也就是给出现一次的次数减一
    p->count++;             //然后给   p->count 置为出现过2 次
    count[p->count]++;
    return true;
}

void T_insert(char *s)
{
    int len=strlen(s);
    Node *p=root;
    for(int i=0;i<len;i++)
    {
        int pos=fun(s[i]);
        if(p->next[pos]==NULL)
        {
            p->next[pos]= new Node(s[i]);
        }
        p=p->next[pos];
    }
    p->count=1;
    count[p->count]++;
}

/*
9 6
AAAAAA
ACACAC
GTTTTG
ACACAC
GTTTTG
ACACAC
ACACAC
TCCCCC
TCCCCC
0 0
*/

int main()
{

    int n,m;
    while(scanf("%d%d",&n,&m)==2&&n&&m)
    {
        count =new int [n+1];
        fill(count,count+n+1,0);
        for(int i=0;i<n;i++)
        {
            char *s;
            s= new char[m+1];
            scanf("%s",s);
            if(T_Find(s)==false)
            {
                T_insert(s);
            }
        }
        int i=0;
        for( i=1;i<=n;i++)
        {
            printf("%d\n",count[i]);
        }

    }
    return 0;
}