uestc1633

uestc1633

题意

给你一个大小为 (n) 的集合 (S) ，集合里有 (n) 个互不相同正整数，有 (q) 个询问，每次询问是否能选择 (S) 中的一些数字 ( 同一个数字可以选择多次,也可以任何数字都不选)，使它们相加的和为 (m) 。

分析

这种题型 竟然 可以套用最短路的模型。

如果 (k) 在集合中，那么如果 (a) 是合法的和的方案，那么 (a + k) 一定是合法的。
那么我们只要求出 (\% k) 后得到 ([0, k - 1]) 这些数的最小的和（ (d) 数组）。那么判断是否可以组成 (m) ，只需要 (m >= d[m \% k]) 。

主要就是维护一个小的先出的优先队列 ，所以先出来的值 ( 假设是 (x) ) (\% k) 的相同的余数中一定是最小的，这个时候就要标记 (vis[x \% k] = 1) ，所有后面出来的值 (y) ，如果 (vis[y \% k]) 已经标记过 ( 假设就是前面的 (x) 标记的 ) ，可以直接跳过，因为 (x \% k == y \% k) ，且 (x < y) ，前面 (x) 出队列后已经更新了其它的可能出现的余数，在拿 (y) 去更新就没必要了。

code

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 2e3 + 10;
const int INF = 1e9 + 7;
int n, a[MAXN];
int d[50005];
int vis[50005];
void dij(int w) {
    priority_queue<int, vector<int>, greater<int> > q;
    fill(d, d + w, INF);
    q.push(0);
    d[0] = 0;
    while(!q.empty()) {
        int now = q.top(); q.pop();
        if(vis[now % w]) continue;
        vis[now % w] = 1;
        for(int i = 0; i < n; i++) {
            int dist = now + a[i];
            if(dist < d[dist % w]) {
                d[dist % w] = dist;
                q.push(dist);
            }
        }
    }
}
int main() {
    scanf("%d", &n);
    int w = INF;
    for(int i = 0; i < n; i++) {
        scanf("%d", &a[i]);
        w = min(w, a[i]);
    }
    dij(w);
    int q;
    scanf("%d", &q);
    while(q--) {
        int m;
        scanf("%d", &m);
        puts(m >= d[m % w] ? "YES" : "NO");
    }
    return 0;
}