【题目链接】
【思路要点】
- 首先,如果问题有解,必定存在一组解为\((a_1,a_2,...,a_N)\),这一点是充要的。
- 充分性显然,对于必要性,我们假设问题有解,令这组解为\((p_1,p_2,...,p_k)\),但\((a_1,a_2,...,a_N)\)不是一组合法的解,即\((a_1,a_2,...,a_N)\)可能组合出\((a_1,a_2,...,a_N)\)以外的\(M\)以内的数字,那么我们用\((p_1,p_2,...,p_k)\)也一定能组合出这个数字,因此\((p_1,p_2,...,p_k)\)不是一组合法的解,原命题成立。
- 从这个证明中,我们也可以看出,原问题无解等价于\((a_1,a_2,...,a_N)\)可以组合出\((a_1,a_2,...,a_N)\)以外的\(M\)以内的数字,这一点我们可以用一次NTT求多项式的平方得到。
- 注意我们只需要判断是否存在这样的数字,而不需要找出所有的这样的数字,因此不需要求解这个多项式的\(M\)次方,只要求它的平方就可以判断了。
- 接下来,我们需要使得问题的解尽可能优。
- 如果一个数\(p_i\)可以被\((p_1,p_2,...,p_k)\)按照不唯一的方式组合出来,换而言之,\(p_i\)可以被\((p_1,p_2,...,p_{i-1})\)组合出来,那么\(p_i\)没有必要出现在答案中,因为\(p_i\)具有等效的替代品。
- 注意到我们现在已经保证了问题有解,也就是\((a_1,a_2,...,a_N)\)不可能组合出\((a_1,a_2,...,a_N)\)以外的\(M\)以内的数字,再换一种说法,假设\((a_1,a_2,...,a_N)\)能够组合出\(X\)并且\(X≤M\)那么\(X\)一定是\((a_1,a_2,...,a_N)\)中的一个,并且如果\(X+a_i≤M\),那么\(X+a_i\)也一定是\((a_1,a_2,...,a_N)\)中的一个。
- 那么实际上,我们只需要在NTT对多项式平方的结果对应位置找到系数为0处输出即可,也即\(a_x\)能作为答案中的一部分当且仅当不存在\(i\),\(j\)使得\(a_i+a_j=a_x\)。
- 时间复杂度\(O(N+MLogM)\)。
【代码】
#include<bits/stdc++.h> using namespace std; const int MAXN = 2097152; const int P = 998244353; const int G = 3; template <typename T> void chkmax(T &x, T y) {x = max(x, y); } template <typename T> void chkmin(T &x, T y) {x = min(x, y); } template <typename T> void read(T &x) { x = 0; int f = 1; char c = getchar(); for (; !isdigit(c); c = getchar()) if (c == '-') f = -f; for (; isdigit(c); c = getchar()) x = x * 10 + c - '0'; x *= f; } template <typename T> void write(T x) { if (x < 0) x = -x, putchar('-'); if (x > 9) write(x / 10); putchar(x % 10 + '0'); } template <typename T> void writeln(T x) { write(x); puts(""); } long long power(int x, int y) { if (y == 0) return 1; long long tmp = power(x, y / 2); if (y % 2 == 0) return tmp * tmp % P; else return tmp * tmp % P * x % P; } bool exist[MAXN]; long long a[MAXN]; int N, Log, cnt, ans[MAXN], home[MAXN]; void NTTinit() { for (int i = 0; i < N; i++) { int ans = 0, tmp = i; for (int j = 1; j <= Log; j++) { ans <<= 1; ans += tmp & 1; tmp >>= 1; } home[i] = ans; } } void NTT(long long *a, int mode) { for (int i = 0; i < N; i++) if (home[i] < i) swap(a[i], a[home[i]]); for (int len = 2; len <= N; len <<= 1) { long long delta; if (mode == 1) delta = power(G, (P - 1) / len); else delta = power(G, P - 1 - (P - 1) / len); for (int i = 0; i < N; i += len) { long long now = 1; for (int j = i, k = i + len / 2; k < i + len; j++, k++) { long long tmp = a[j]; long long tnp = a[k] * now % P; a[j] = (tmp + tnp) % P; a[k] = (tmp - tnp + P) % P; now = now * delta % P; } } } if (mode == -1) { long long inv = power(N, P - 2); for (int i = 0; i < N; i++) a[i] = a[i] * inv % P; } } int main() { int n, m; read(n), read(m); for (int i = 1; i <= n; i++) { int x; read(x); a[x] = exist[x] = 1; } N = 1, Log = 0; while (N <= 2 * m) { N <<= 1; Log++; } NTTinit(); NTT(a, 1); for (int i = 0; i < N; i++) a[i] = a[i] * a[i] % P; NTT(a, -1); for (int i = 1; i <= m; i++) if (a[i] && !exist[i]) { printf("NO\n"); return 0; } else if (!a[i] && exist[i]) ans[++cnt] = i; printf("YES\n%d\n", cnt); for (int i = 1; i <= cnt; i++) printf("%d ", ans[i]); return 0; }