EduRound 126 (Rated for Div. 2)C题

  没想出来，看了别人的思路，感觉二分更好理解，代码写的比较low，记录一下。。。

题目：C. Water the Trees

题意：
  给一些树浇水，奇数天可使树长高1，偶数天可使树长高2，也可以跳过一天不浇水，给出初始的树的高度，问最少需要多少天才能使所有树一样高。

思路：
  1、最终树的高度一定是原来的最大值或者最大值加1。
  2、二分答案ans求最小值，则有ans/2个奇数天(odd)和偶数天(even)，即+1和+2的次数(ans为奇数时+1的次数为ans/2+1)，然后检查是否可行，总差值为sum，差值为奇数的个数为cnt，若cnt <= odd 且 sum - cnt <= odd- cnt + 2 * even，则可行。

代码：

#include <iostream>
#include <algorithm>
#include <queue>
#define int long long

using namespace std;

const int N = 3e5+10;
int t, n;
int h[N];

bool check(int sum, int cnt, int mid)
{
    int odd = mid / 2, even = mid / 2;
    if(mid & 1)
        odd++;
    if(odd < cnt)
        return false;
    if(sum-cnt <= odd-cnt + even * 2)
        return true;
    return false;
}

int solve(int mmax)
{
    int sum = 0, cnt = 0;
    for(int i = 0; i < n; i++)
    {
        sum += (mmax-h[i]);
        if((mmax - h[i]) & 1)
            cnt++;
    }
    int l = 0, r = sum * 2 - 1;
    while(l < r)
    {
        int mid = l + r >> 1;
        if(check(sum, cnt, mid))
            r = mid;
        else
            l = mid + 1;
    }
    return l;
}

signed main()
{
    cin >> t;
    while(t--)
    {
        cin >> n;
        int mmax = -1;
        for(int i = 0; i < n; i++)
        {
            cin >> h[i];
            mmax = max(mmax, h[i]);
        }
        int res = min(solve(mmax), solve(mmax+1));
        cout << res << endl;
    }
    return 0;
}