第八屆藍橋杯 java B組 包子湊數

題目：包子湊數

小明幾乎每天早晨都會在一家包子鋪吃早餐。他發現這家包子鋪有N種蒸籠，其中第i種蒸籠恰好能放Ai個包子。每種蒸籠都有非常多籠，可以認為是無限籠。 每當有顧客想買X個包子，賣包子的大叔就會迅速選出若幹籠包子來，使得這若幹籠中恰好一共有X個包子。比如一共有3種蒸籠，分别能放3、4和5個包子。當顧客想買11個包子時，大叔就會選2籠3個的再加1籠5個的（也可能選出1籠3個的再加2籠4個的）。 當然有時包子大叔無論如何也湊不出顧客想買的數量。比如一共有3種蒸籠，分别能放4、5和6個包子。而顧客想買7個包子時，大叔就湊不出來了。 小明想知道一共有多少種數目是包子大叔湊不出來的。 輸入 ---- 第一行包含一個整數N。(1 <= N <= 100) 以下N行每行包含一個整數Ai。(1 <= Ai <= 100)   輸出 ---- 一個整數代表答案。如果湊不出的數目有無限多個，輸出INF。 例如， 輸入： 2   4   5    程式應該輸出： 6   再例如， 輸入： 2   4   6     程式應該輸出： INF 樣例解釋： 對于樣例1，湊不出的數目包括：1, 2, 3, 6, 7, 11。   對于樣例2，所有奇數都湊不出來，是以有無限多個。   資源約定： 峰值記憶體消耗（含虛拟機） < 256M CPU消耗  < 1000ms

請嚴格按要求輸出，不要畫蛇添足地列印類似：“請您輸入...” 的多餘内容。

貼上一位大神寫的代碼 http://blog.csdn.net/sangjinchao/article/details/70170752

以下是我寫的。

不知道有沒有錯。

import java.util.Arrays;
import java.util.Scanner;

class Main {
	static int N, max = 10010;
	static boolean[] vis = new boolean[max];
	static int[] a;

	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		N = in.nextInt();
		a = new int[N];
		for (int i = 0; i < N; i++)
			a[i] = in.nextInt();
		Arrays.sort(a);
		for (int i = 0; i < N; i++)
			for (int j = a[i]; j < max; j += a[i]) {
				vis[j] = true;
				for (int k = 0; k < N; k++) {
					if (j + a[k] >= max)
						break;
					vis[j + a[k]] = true;
				}
			}
		for (int i = 1; i < max; i++) {
			if (!vis[i])
				continue;
			for (int j = 1; j < max; j++) {
				if (!vis[j])
					continue;
				if (i + j < max)
					vis[i + j] = true;
			}
		}
		int count = 0;
		int flag = 0;
		for (int i = 1; i < max; i++) {
			if (vis[i])
				flag++;
			if (flag == a[0]) {
				System.out.println(count);
				return;
			}
			if (!vis[i]) {
				count++;
				flag = 0;
			}
		}
		System.out.println("INF");
	}
}