ZOJ ACM 1204 (JAVA)

畢業好幾年了，對算法仍是比較有興趣，因此想又一次開始作ACM題。俺作題比較任意，通常先挑經過率高的題來作。

第1204題，詳細描寫敘述請參考，ZOJ ACM 1204

1）難度分析

這個題目，基本的難度在於要依據長度來排序。

比方1 2 3 4 5 6。結果必須爲:

1+2=3
1+3=4
1+4=5
1+5=6
2+3=5
2+4=6
1+2+3=6

但是個人結果爲：

1+2=3

1+2+3=6

1+3=6

。。。

2）解決方法

將結果緩存，依據長度排序後再打印。

3）AC的源代碼。假設要直接提交，請將中文凝視刪除。

import java.util.Collections;  
import java.util.Comparator;  

public class Main {
	static int lineNumber;
	static int input[];
	static int M = 4;
	static boolean hasR = false;
	static java.util.ArrayList<int []> results;

	public static void main(String[] args) {

<span style="white-space:pre">		</span>//排序算法重寫
		Comparator<int []> comparator = new Comparator<int []>(){  
			   public int compare(int []r1, int []r2) {  

			    if(r1[0] != r2[0]){  
			     return r1[0]-r2[0];  
			    }
			    else {
			    	return 0;
			    }
			   }
		};  
		
		java.util.Scanner scanner = new java.util.Scanner(System.in);
		
		if(scanner.hasNext()) {
			lineNumber = Integer.parseInt(scanner.nextLine());
		}
		int lineIndex = 0;
		while(scanner.hasNext()) {
			lineIndex++;
			hasR = false;
			
			String lineStr = scanner.nextLine();
			
			String strs[] = lineStr.split(" ");
			M = Integer.parseInt(strs[0]);
			input = new int[M];
			results = new java.util.ArrayList<int []>();
			
			for(int j=0; j<M;j++) {
				input[j] = Integer.parseInt(strs[j+1]);
			}
			java.util.Arrays.sort(input);       //<span style="font-family: Arial, Helvetica, sans-serif;">//說明：比較關鍵，因爲一開始沒有注意看題，沒有考慮輸入的一行數字多是無序的，直接依照升序處理。</span>


			fn(0,0,0,new int[33]);
			
			if(hasR == false) {
				System.out.println("Can't find any equations.");
			} else {
				Collections.sort(results,comparator);

				for(int i = 0;i<results.size();i++) {
					int count = results.get(i)[0];
					for(int j=1;j<=count;j++) {
						System.out.print(results.get(i)[j]);
						if(j<count){
							System.out.print("+");
						}
					}
					System.out.println("="+results.get(i)[count+1]);
				}
			}
			
			System.out.println("");
			if(lineIndex == lineNumber) {
				break; 
			}
		}
	}
	
	public static long fn(int i, int sum, int count, int[] res) {
		long r;
		
		if(i==M)return -1;
		if(sum >input[M-1]) return -1;            //說明：比較關鍵。直接決定了算法的時間複雜度，沒有這個推斷。計算超時。


		if (sum == input[i]) {
			hasR = true;
			res[count+1] = sum;
			results.add(res);
		}

		if (i < M - 1) {
			int res_new[] = new int[33];
			for (int k = 0; k <= count + 1; k++) {
				res_new[k] = res[k];
			}
			res_new[0] = count + 1;
			res_new[count + 1] = input[i];
			fn(i + 1, sum + input[i], count + 1, res_new);

			fn(i + 1, sum, count, res);
		}
		
		return -1;
		
	}
}