POJ2001--Shortest Prefixes

http://poj.org/problem?id=2001

本題就是關於前綴的,可以建一個樹,每個節點有個int來記錄經過此節點單詞的個數。如果此int爲零就可以作爲前綴,從根節點出發,第一個爲1的就是最短的前綴.

在做此題是糾結的一點就是,diverge到底放在哪裏,或者說插入一個單詞後,怎麼增加diverge,哪些需要增加?沒有考慮清楚我就寫了。花了很長時間。

最終決定:插入一個單詞,每個路徑上diverge都+1,包括最終的地方(diverge初始值是0)。根節點不像二叉樹起始是空,而是來了一個實體,其26個指針爲空。原本我還準備在結構體里加一個char,來只是當前字母,後來發現沒意義。

另外注意insert函數遞歸出口。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>

typedef struct node{
	node *next[26];
	int diverge;
	bool isword;
}Word;

Word *words;

void insert(char* s,Word* w){
	//單詞末尾,遞歸出口
	if(s[0]=='\0'){
		w->diverge++;
		//isword後來沒有用上
		w->isword=true;
		return ;
	}
	//指示爲空需新建節點
	if(w->next[s[0]-'a']==NULL){
		Word *temp=(Word*)malloc(sizeof(Word));
		temp->diverge=0;
		int i;
		for(i=0;i<26;i++){
			temp->next[i]=NULL;
		}
		w->next[s[0]-'a']=temp;
	}
	//不空,注意沒有else,因爲if裏面也要到這裏進行處理
	w->next[s[0]-'a']->diverge++;
	//插入下一個字母
	insert(s+1,w->next[s[0]-'a']);
	
}

int main(){
	words = (Word*)malloc(sizeof(Word));
	words->diverge=0;
	words->isword=false;
	int i=0;
	for(i=0;i<26;i++){
		words->next[i]=NULL;
	}
	i=0;
	char str[1000][21];
	while(scanf("%s",str[i])!=EOF){
		insert(str[i],words);
		i++;
	}
	int num=i;
	for(i=0;i<num;i++){
		int len=strlen(str[i]);
		int j=0;
		//輸出單詞
		printf("%s ",str[i]);
		Word* temp=words;
		for(j=0;j<len;j++){
			//diverge不是1,輸出字符,直到結束或diverge爲1.
			printf("%c",str[i][j]);////
			temp=temp->next[str[i][j]-'a'];
			if(temp->diverge==1)
				break;
		}
		puts("");
	}
	
	return 0;
}



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章