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;
}