蓝桥杯-prev8-买不到的数目

问题描述

小明开了一家糖果店。他别出心裁：把水果糖包成4颗一包和7颗一包的两种。糖果不能拆包卖。
小朋友来买糖的时候，他就用这两种包装来组合。当然有些糖果数目是无法组合出来的，比如要买 10 颗糖。
你可以用计算机测试一下，在这种包装情况下，最大不能买到的数量是17。大于17的任何数字都可以用4和7组合出来。
本题的要求就是在已知两个包装的数量时，求最大不能组合出的数字。

输入格式

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

输出格式

一个正整数，表示最大不能买到的糖数

样例输入1

4 7

样例输出1

17

样例输入2

3 5

样例输出2

7

bfs加使用数组标记，两个数都是1000以内，所以结果必然不会太大。
思路就是：从0开始，每次从优先队列里拿出最小值x出来，然后将x+a和x+b入队，并做标记
数组开到1e6，如果x+a或者x+b大于1e6就不入队，最后从大到小遍历一下标记数组，找到第一个未被标记的数即是答案

#include <iostream>
#include <cstdio>
#include <queue>
#define fin freopen("in.txt", "r", stdin)
#define fout freopen("out.txt", "w", stdout)
using namespace std;
const int maxn = 1e6+5;
bool vis[maxn];
int bfs(int a, int b) {
    priority_queue<int, vector<int>, greater<int> > Q;  //维护一个最小值优先队列
    Q.push(0);
    while(!Q.empty()) {
        int x = Q.top();
        Q.pop();
        if(x + a < maxn && !vis[x + a]) Q.push(x + a), vis[x + a] = true;
        if(x + b < maxn && !vis[x + b]) Q.push(x + b), vis[x + b] = true;
    }
    for(int i = maxn - 1; i >= 0; i --) if(!vis[i]) return i;
}

int main() {
    //fin;
    int a, b;
    cin >> a >> b;
    if(a > b) swap(a, b);
    cout << bfs(a, b) << endl;
    return 0;
}