题目:包子凑数
小明几乎每天早晨都会在一家包子铺吃早餐。他发现这家包子铺有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");
}
}