【CodeForces】1006 - C - Three Parts of the Array（思路）

题目链接

题意很简单，只要保证sum1 == sum3,而且sum1为能取到的最大值。

我们假设sum1+sum2+sum3 == sum, 那么会发现sum1<= sum/2, 我们将数列由前
向后求sum1，当sum1 >= sum/2的时候停止并记录此时的位置，此时sum3 == sum - sum1,如果此时sum1 == sum3直接输出。不然sum1和sum3 那个数大就减去标记位置的数直到相等。

AC代码：

#include<bits\stdc++.h>
using namespace std;
const int maxn = 3e5+10;
typedef long long ll;
ll a[maxn];

int main(void)
{
    ll n,sum1 = 0,sum2 = 0,sum = 0,pos1,pos2;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
        sum += a[i];
    }
    for(int i=0;i<n;i++){
        sum1 += a[i];
        if(sum1 >= (sum+1)/2){    //标记sum1和sum3的位置
            pos1 = i;
            pos2 = i+1;
            break;
        }
    }
    sum2 = sum-sum1; 
    if(sum1 == sum2){    //特判
        cout<<sum1<<endl;
        return 0;
    }
    while(sum1 != sum2){    
        if(sum1 > sum2){
            sum1 -= a[pos1];
            pos1--;
        }
        if(sum1 < sum2){
            sum2 -= a[pos2];
            pos2++;
        }
    }
    cout<<sum1<<endl;
    return 0;
}