LeetCode : Unique Binary Search Tree

//先對n個數進行全排列,再對排列出來的每個結果進行判斷是否爲二叉搜索樹(可以將排列結果作爲二叉搜索樹的先序遍歷或者後續遍歷)
//但這種方法在時間上超時
//正確方法應用用卡特蘭數來解
import java.util.Scanner;

public class UniqueBinarySearchTree {
	int count = 0;

	/**
	 * @param args
	 */
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner scan = new Scanner(System.in);
		int n = scan.nextInt();
		int numbers[] = new int[n];
		int i = 0;
		for (; i < n; i++) {
			int temp = scan.nextInt();
			System.out.println(temp);
			numbers[i] = temp;
		}
		System.out.println("i= " + i);
		UniqueBinarySearchTree ubst = new UniqueBinarySearchTree();
		ubst.binarysearch(numbers, 0, n);
		System.out.println("count=  " + ubst.count);
	}

	public void binarysearch(int num[], int start, int n) {
		if (start == n) {
			boolean result = isbinarysearchtree(num, n);
			if (result == true) {
				count++;
			}
			return;
		}

		for (int i = start; i < n; i++) {
			swap(num, i, start);
//			System.out.println(num[i] + " " + num[start]);
			binarysearch(num, start + 1, n);
			swap(num, i, start);
		}
	}

	public void swap(int num[], int a, int b) {
		int temp = num[a];
		num[a] = num[b];
		num[b] = temp;
	}

	public boolean isbinarysearchtree(int[] result, int n) {
		if (result == null || n == 0) {
			return false;
		}
		int root = result[n - 1];
		int i;
		for (i = 0; i < n-1; i++) {
			if (result[i] > root) {
				break;
			}
		}
		int j = i;
		for (; j < n-1; j++) {
			if (result[j] < root) {
				return false;
			}
		}
		boolean left = true;
//		System.out.println("i= "+i);
		if (i > 0) {
			left = isbinarysearchtree(result, i);
		}
		boolean right = true;
		if (i < n - 1) {
			right = isbinarysearchtree(Arrays.copyOfRange(result, i, n-1), n -i- 1);
		}
	
		return left&&right;
	}

}

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