南昌大学航天杯第二届程序设计竞赛校赛网络同步赛 G

链接：https://www.nowcoder.com/acm/contest/122/G
来源：牛客网

题目描述

阿汤同学为了准备下学期的 ACM-ICPC，刷了很多的题目，他觉得自己已经比较厉害了，于是想出个题目考考你。现在他给你一个数组 A，问你是否能将该数组划分成数组 B、C 使得 B 数组的平均数和C 数组的平均数相等，数组 B 和 C 都不能为空。

输入描述:

从标准输入读入数据。
输入包含多组数据，第一行一个整数 T 代表数据组数。接下来依次描述每组数据，对于每组数据：
第一行输入正整数 N，第二行输入 N 个非负整数

1≤|A|≤30 （数组 A 的长度范围在 1 到 30 之间 ）
0≤A[i]≤10000 （数组 A 中的元素）

输出描述:

输出到标准输出。
对于每组数据，输出一行：
如果能划分成满足题目要求的数组 B 和 C 则输出 yes，否则输出no

示例1

输入

1
8
1 2 3 4 5 6 7 8

输出

yes

说明

样例说明：将数组 A 划分成【1,4,5,8】和【2,3,6,7】,平均数为 4.5

#include<bits/stdc++.h>
using namespace std;
#define MAXN 2001000
#define PSIZE 100000
bool Average(vector<int>& A) {
        static bool dp[2][16][300001];
        int sum = 0;
        for (int a: A) sum += a;

        int N = A.size();
        memset(dp, 0, sizeof(dp));
        for (int n = 0; n <= N; n++) {
            dp[n & 1][0][0] = true;
        }
        for (int n = 1; n <= N; n++) {
            for (int k = 1; k <= N/2; k++) {
                for (int s = 1; s <= sum; s++) {
                    if (s >= A[n - 1]) {
                        dp[n & 1][k][s] = dp[n-1 & 1][k][s] || dp[n-1 & 1][k-1][s - A[n - 1]];
                    } else {
                        dp[n & 1][k][s] = dp[n-1 & 1][k][s];
                    }
                }
            }
        }
        for (int k = 1; k <= N/2; k++) {
            if ((k * sum / N * N == k * sum) && dp[N & 1][k][k * sum / N]) return true;
        }
        return false;
    }
int main(){
    int t;
    while(cin>>t){
        while(t--){
                vector<int>Ve;
                int n;
                int num;
                cin>>n;
                for(int i=0;i<n;i++){
                    cin>>num;
                    Ve.push_back(num);
                }
                if(Average(Ve)){
                    cout<<"yes"<<endl;
                }else{
                    cout<<"no"<<endl;
                }
        }
    }
    return 0;
}