蓝桥杯-历届试题 包子凑数

问题描述

小明几乎每天早晨都会在一家包子铺吃早餐。他发现这家包子铺有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

---

写在题解前：这道题实际上就是动态规划的题，我自己写了一种解法，后来又找到另外一种解法，不过两种解法的其实大同小异，只不过标记的方式不一样，可以看看作为思路扩展
第一种解法：

import java.util.*;

public class Main{  
		
    public static void main(String[] args) {
    	Scanner sc =new Scanner(System.in);
    	int N;
    	N = sc.nextInt(); // 输入蒸笼种类数
    	int[] a = new int[N]; // 各种蒸笼可放的包子数
    	int[] t = new int[1000000]; // 动态规划数组
    	for(int i=0;i<N;i++){
    		a[i] = sc.nextInt();
    	}
    	// 判断所有蒸笼可放的包子数是否有非1的公约数，
    	// 有则说明没办法凑出数目是无限的，这个脑补一下应该就理解了
    	if(!valid(a)){ 
    		System.out.println("INF");
    	}
    	else {
    		t[0]=1;
        	for(int i=0;i<N;i++){
        		for(int j=a[i];j<1000000;j++){
        		 // 这里的判断是为了防止做好的标记被重新覆盖掉，
        		 // 只对那些还没凑出来的数目进行判断
        			if(t[j]!=1)
        			t[j]=t[j-a[i]];
        		}
        	}
        	int sum=0;
        	for(int i=0;i<1000000;i++) // 统计没办法凑出来的数目数量
        		if(t[i]!=1)
        			sum+=1;
        	System.out.println(sum);
    	}
    }
    // 判断数组temp中的元素是否有非1的公约数
    public static boolean valid(int[] temp){ 
    	boolean flag=true;
    	int min = temp[0];
    	for(int i=1;i<temp.length;i++){
    		min = gcd(min,temp[i]);
    	}
    	return min == 1;
    }
	private static int gcd(int a, int b) { // 求两个数的最大公约数
		int i;
		int result=1;
		for(i=1;i<=a&&i<=b;i++){
			if(a%i==0&&b%i==0)
				result=i;
		}
		return result;
	}
}

第二种解法：

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

public class Main{
	public static void main(String[] args){
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt(); // 输入蒸笼种类数
		int[] s = new int[n]; // 各种蒸笼可放的包子数
		
		for(int i=0;i<n;i++){ 
			s[i] = sc.nextInt();
		}
		
		// 判断所有蒸笼可放的包子数是否有非1的公约数
		// 有则说明没办法凑出数目是无限的，这个脑补一下应该就理解了
		if(test(s)){
			System.out.println("INF");
		} else {
			int[] t = new int[1000000];
			for(int i=0;i<n;i++){ // 先标记最基本的可凑出来的数目
				t[s[i]] = 1;
			}
			t[0]=1;
			for(int i=1;i<1000000;i++){
				for(int j=0;j<n;j++){
					// 这里不大会说，不过应该看得懂
					if(t[i]==1&&i+s[j]<1000000  ){ 
						t[i+s[j]] = 1;
					}
				}
			}
			int sum = 0;
			 // 统计没办法凑出来的数目数量
			for(int i=1;i<1000000;i++) {
				if(t[i]==0)
					sum++;
			}
			System.out.println(sum);
		}
		
	}

	private static boolean test(int[] s) { // 判断数组temp中的元素是否有非1的公约数
		int a = s[0];
		for(int i=1;i<s.length;i++){ 
			a = gcd(a,s[i]);
		}
		return a!=1;
	}

	private static int gcd(int a, int b) { // 求两个数的最小公约数
		for(int i=2;i<=a&&i<=b;i++){
			if(a%i==0&&b%i==0)
				return i;
		}
		return 1;
	}
}