蓝桥杯 买不到的数目

小明开了一家糖果店。他别出心裁：把水果糖包成4颗一包和7颗一包的两种。糖果不能拆包卖。

    小朋友来买糖的时候，他就用这两种包装来组合。当然有些糖果数目是无法组合出来的，比如要买 10 颗糖。

    你可以用计算机测试一下，在这种包装情况下，最大不能买到的数量是17。大于17的任何数字都可以用4和7组合出来。

    本题的要求就是在已知两个包装的数量时，求最大不能组合出的数字。

输入：
两个正整数，表示每种包装中糖的颗数(都不多于1000)

要求输出：
一个正整数，表示最大不能买到的糖数

例如：
用户输入：
4 7
程序应该输出：
17

再例如：
用户输入：
3 5
程序应该输出：

7

思路：这道题可以用递推来做，用数组arr[i]=0表示数字i能通过这两个数a,b(a<b)组合得到，arr[i]=1表示组合不到，答案就是求a[i]=1的最大值。arr[i]可以根据arr[i-a]和arr[i-b]求出。如果arr[i-a]或者a[i-b]有一个能够通过a,b组合得到即arr[i-a]或arr[i-b] 至少有一个等于0。 arr[i]=1的最大值就是当从i-a+1开始,如果连续a个数都能够通过a,b组合得到，则n>=i的数都可以组合得到。

#include<iostream>
#include<cstring>
using namespace std;
int arr[1000000];
int main(){
	//arr[i]==0 表示数量是i的时候可以换 
	int a,b,i;
	memset(arr,0,sizeof(arr)); 
	cin>>a>>b;
	if(a>b){
	    i=a;
		a=b;
		b=i;
	}
	i=1;
	int k=0;
	while(1){
		if(i<a){
			arr[i]=1;
			
		}else if(i>=a&&i<b){
			if(arr[i-a]!=0){ //买不到 
				arr[i]=1;
				k=0;
			}else{          //买得到 
				k++;
				if(k==a)
				break;
			}
		}else{
			if((arr[i-a])!=0&&(arr[i-b]!=0)){
				arr[i]=1;
				k=0;
			}else{
				k++;
				if(k==a)
				break;
			}
		}
		i++;
	}
	cout<<i-k<<endl;
	return 0;
}

 
资源约定：
峰值内存消耗 < 64M
CPU消耗  < 3000ms