题意
有编号1-n的n(1 <= n <= 50000)个格子,机器人从1号格子顺序向后走,一直走到n号格子,并需要从n号格子走出去。机器人有一个初始能量,每个格子对应一个整数A[i](-1000000000 <= A[i] <= 1000000000),表示这个格子的能量值。如果A[i] > 0,机器人走到这个格子能够获取A[i]个能量,如果A[i] < 0,走到这个格子需要消耗相应的能量,如果机器人的能量 < 0,就无法继续前进了。问机器人最少需要有多少初始能量,才能完成整个旅程。
例如:n = 5。{1,-2,-1,3,4} 最少需要2个初始能量,才能从1号走到5号格子。途中的能量变化如下3 1 0 3 7。
解题思路
二分查找这个初始能量
参考代码
#include <iostream>
#include <string.h>
using namespace std;
typedef long long ll;
#define MAXN 50000+5
ll INF = 1e17;
int a[MAXN],n;
bool success(ll x){
for (int i=0;i<n;i++){
x+=a[i];
if (x<0)
return 0;
}
return 1;
}
int main(){
while (cin>>n){
for (int i=0;i<n;i++)
cin>>a[i];
ll l=0,r=INF;
int flag=0;
while (l<r){
ll mid=(l+r)/2;
if (success(mid)==1) r=mid-1;
else l=mid+1;
if (success(l)==0&&success(r)==0){
cout<<mid<<endl;
flag=1;
break;
}
}
if (flag==0) cout<<l<<endl;
}
return 0;
}