Codeforces Round #493 (Div. 2) ---- A Balloons

给了我们一些包裹,每个包裹里有一些气球,怎么拿可以满足题目中给我们的条件。

我们直接枚举第一个人拿了多少个,然后总个数减去第一个人拿掉的就是剩下的个数。

注意第一个人不能把所有的气球都全部拿掉。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn = 10000;
int a[maxn],sum[maxn];
int main(){
    int n,all = 0;
    cin >> n;
    for (int i=1; i<=n; i++) {
        cin >> a[i];
        sum[i] = sum[i-1] + a[i];
        all += a[i];
    }
    for (int i=1; i<n; i++) {
        if (sum[i] != all - sum[i]) {
            cout << i << endl;
            for (int j=1; j<=i; j++) {
                cout << j << ' ';
            }
            cout << endl;
            return 0;
        }
    }
    cout << -1 << endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/CCCCTong/article/details/81060286