ホームページ > 記事 > ウェブフロントエンド > Codeforces ベータ ラウンド #4 (ディビジョン 2 のみ) C. 登録 system_html/css_WEB-ITnose
この質問は、以前は辞書ツリーに書くのが最も簡単だったのですが、過去数回のコンテストでは、質問の主なアイデアは
でした。 いくつかの文字列を指定して、各文字列をクエリします。表示されない場合は、新しい文字列を生成します。この文字列が何回表示されるかを返します。 。
ソリューションのアイデア:
辞書ツリー、各文字列の挿入数をカウントします。
コードは次のとおりです:
#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;}