D.数组成鸡(2024牛客寒假训练营)

题目大意：经过多次操作(数组全＋1/-1)后，能否使数组所有数的乘积等于整数M

思路解析：

枚举有限次加一减一操作组成答案集，看所询问的M是否在答案集中。关键要对某些不必要的操作进行剪枝处理。

我们可以用一个set容器处理一下重复的元素，当该数组里互不相同的元素<=16的时候才在我们的考虑范围之内，因为最好的一种情况就是(8!)^2>1e9超过了M的范围。可以对数组里的元素从小到大排个序，然后枚举答案，重点关注开头两个元素和末尾两个元素，不管是开头还是末尾，其两个元素的都要小于 。最后留一个问题供大家思考，0为啥得要特殊考虑一下？

废话不多说，代码如下：

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
using T = pair<int, int>;
//set<int>S;
//unordered_map<int, int>mp;
const int N = 2e5 + 10;
ll a[N];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    int n, Q; cin >> n >> Q;
    ll i,j;
    set<ll>ans;
    set<ll>st;
    ans.insert(0);//0可能因枚举答案时开头或末尾抑或是中间某一部分超出M的范围而没被记录
    for (i = 1; i <= n; ++i) {
        cin >> a[i];
        st.insert(a[i]);
    }
    sort(a + 1, a + n + 1);
    ll tmp = a[1];
    if (st.size() <= 16) {
        for (i = 1; i <= n; ++i)a[i] -= tmp;
        tmp = a[2];
        for (i = -1e5-tmp; i <= 1e5&&a[n-1]+i<1e5; ++i) {
            if(a[1]+i<-1e9)continue;
            ll res = 1;
            int flag = 1;
            for (j = 1; j <= n; ++j) {
                res *= (a[j] + i);
                if (abs(res) > 1e9) {
                    flag = 0; break;
                }
            }
            if (flag)
                ans.insert(res);
        }
    }
    for (i = 1; i <= Q; ++i) {
        ll num; cin >> num;
        if (ans.count(num)) {
            cout << "Yes" << endl;
        }
        else cout << "No" << endl;
    }
    return 0;
}