2021 RoboCom 世界機器人開發者大賽-本科組（初賽）7-1 懂的都懂 (20 分)

 

 賽場上我暴力做的，但是先做的后面的題。。。做這個題的時候時間不夠了，腸子都悔青了，最后一次提交的時候已經只剩一分鍾，測試點0沒過，十分沒了，試題開放后，就改了一點測試點0就通過了。。。測試點0就是浮點運算的考慮，不要計算平均數，把要測試的圖像乘四就行，不然會有浮點偏差，啊啊啊啊啊啊！

C++

#include <iostream>
#include <set>
#include <vector>

using namespace std;

int main() {
    int N = 0, K = 0;
    cin >> N >> K;
    vector<int> imag_src(N);
    set<int> avg;
    for (int i = 0; i < N; ++i)
        cin >> imag_src[i];
    for (int a = 0; a < N; ++a) {
        for (int b = 0; b < N; ++b) {
            for (int c = 0; c < N; ++c) {
                for (int d = 0; d < N; ++d) {
                    int tmp = 0;
                    if (a!=b && a!= c && a != d && b != c && b != d && c != d) {
                        tmp = (imag_src[a] + imag_src[b] + imag_src[c] + imag_src[d]);
                        avg.insert(tmp);
                    }
                }
            }
        }
    }

    for (int i = 0; i < K; ++i) {
        int Mi = 0;
        cin >> Mi;
        vector<int> img;
        for (int j = 0; j < Mi; ++j) {
            int img_tmp;
            cin >> img_tmp;
            img.push_back(img_tmp);
        }
        bool flag = true;
        for (int j = 0; j < img.size(); ++j) {
            if(avg.find(img[j] * 4) == avg.end()) {
                flag = false;
                break;
            }
        }
        if(flag) cout << "Yes" << endl;
        else cout << "No" << endl;
    }


    return 0;
}