Codeforces Beta Round #4 (Div. 2 Only) C. Registration system

这个题感觉还不错，以前字典树写的是最顺手的，这几次比赛屡屡挂在字典树上也是有阴影了啊~~

题目大意：

给出一些字符串，对每个字符串进行查询，若没出现过返回OK，若出现过就生成新字符串,格式为原字符串+数，数为这个字符串第几次重复出现。

解题思路：

字典树，对于每个字符串的插入次数进行计数。

下面是代码：

#include <set>
#include <map>
#include <queue>
#include <math.h>
#include <vector>
#include <string>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <cctype>
#include <algorithm>

#define eps 1e-10
#define pi acos(-1.0)
#define inf 107374182
#define inf64 1152921504606846976
#define lc l,m,tr<<1
#define rc m + 1,r,tr<<1|1
#define zero(a) fabs(a)<eps
#define iabs(x)  ((x) > 0 ? (x) : -(x))
#define clear1(A, X, SIZE) memset(A, X, sizeof(A[0]) * (min(SIZE,sizeof(A))))
#define clearall(A, X) memset(A, X, sizeof(A))
#define memcopy1(A , X, SIZE) memcpy(A , X ,sizeof(X[0])*(SIZE))
#define memcopyall(A, X) memcpy(A , X ,sizeof(X))
#define max( x, y )  ( ((x) > (y)) ? (x) : (y) )
#define min( x, y )  ( ((x) < (y)) ? (x) : (y) )

using namespace std;

struct node
{
    int p;
    int num[26];
}node[500000];

char s[50];

int cnt,len;

int InsertString(int strp,int p)
{
    if(strp==len)
    {
        node[p].p++;
        return node[p].p;
    }
    if(node[p].num[s[strp]-'a']==0)
    {
        node[p].num[s[strp]-'a']=cnt++;
    }
    return InsertString(strp+1,node[p].num[s[strp]-'a']);
}
int main()
{
    int n,temp;
    cnt=1;
    clearall(node,0);
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%s",s);
        len=strlen(s);
        temp=InsertString(0,0);
        if(temp==1)
        {
            puts("OK");
        }
        else printf("%s%d\n",s,temp-1);
    }
    return 0;
}