F.Pulling Their Weight(数学思维&均分两组的分界只)(2021年度训练联盟热身训练赛第一场)

传送门

题意：寻找一个中间值 t ，使得小于t的所有数的和与大于t的所有数的和能够相等（至于等于 t 的数则想办法均分），保证存在这样的 t 值。

思路：题目说了所有数不会超过20000，那么我们就从小枚举 t 值，且在过程中以前求小于 t 的数和进行判断即可。

代码实现：

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x3f3f3f3f3f3f3f3f;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9+7;
const int  N = 2e5 + 5;

inline void read(int &x){
    char t=getchar();
    while(!isdigit(t)) t=getchar();
    for(x=t^48,t=getchar();isdigit(t);t=getchar()) x=x*10+(t^48);
}

int n, a[N], sum, res, ans;
map<int, int> mp;

signed main()
{
    IOS;

    cin >> n;
    for(int i = 1; i <= n; i ++){
        cin >> a[i];
        sum += a[i];
        mp[a[i]] ++;
    }
    for(int t = 1; t <= 20000; t ++){
        if(mp[t]){
            res += mp[t]*t;  //对当前t值以前的数进行求和
            if(res*2==sum){
                ans = t+1;  //如果刚好能均分两组，则选择当前t+1值为分界
                break;
            }
            if(res*2>sum){
                ans = t;   //如果当前组求和已经超过半值，则选择当前t值为分界
                break;
            }
        }
    }
    cout << ans << endl;

    return 0;
}