# Array# Sorting# Lookup

f675 - FJCU_109_Winter_Day2_Lab3 二元搜尋樹

🔗 前往 ZeroJudge 原題

題目描述

題目要求讀取 N 個介於 1 到 100 之間的數字,將這些數字由小到大排序後輸出,然後判斷一個給定的數字 q 是否存在於這 N 個數字中,並輸出 "Yes" 或 "No"。雖然題目名稱是二元搜尋樹,但實際上程式碼並未建立二元搜尋樹。

解題思路

程式碼使用了一個大小為 101 的整數陣列 a 作為雜湊表,用於記錄輸入的數字。陣列的索引代表數字本身,陣列的值 1 表示該數字存在。程式首先讀取 N 個數字,並將對應陣列位置的值設為 1。然後,程式遍歷陣列,將存在(值為 1)的索引(數字)從小到大輸出。最後,程式檢查數字 q 是否在陣列中存在,如果存在則輸出 "Yes",否則輸出 "No"。

複雜度分析

  • 時間複雜度: O(N + 100) (N 是輸入數字的數量,100 是陣列的大小。遍歷輸入數字 O(N),遍歷陣列輸出 O(100))
  • 空間複雜度: O(101) (使用固定大小的陣列,空間複雜度為常數)

程式碼

#pragma GCC optimize("Ofast,unroll-loops,no-stack-protector,fast-math")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#include <stdio.h>
inline int read(){
	int a(0);
	char c('0');
	while(c>='0'){
		a=(a<<3)+(a<<1)+c-'0';
		c=getchar_unlocked();
	}
	return a;
}inline void write(int x) {
	if(x==0)
		putchar_unlocked('0');
	else{
		int stk[9],*ptr(&stk[0]);
		while(x){*ptr=x%10;x/=10;++ptr;}
		while(--ptr>=(&stk[0])){putchar_unlocked(*ptr+'0');}
	}
}
int a[101];
int main(){
	int n=read(),x;
	while(n--){
		x=read();
		a[x]=1;
	}
	x=read();
	for(int i=0;i<101;++i){
		if(a[i]){
			write(i);
			putchar_unlocked('\n');
		} 
	}
	if(a[x]){
		putchar_unlocked('Y');
		putchar_unlocked('e');
		putchar_unlocked('s');
	}
	else{
		putchar_unlocked('N');
		putchar_unlocked('o');
	}
}

Discussion