codeforces round #822 D

Problem - D - Codeforces

题意：

给出一个序列a,你所在的初始位置是k，可以向左或者是向右移动，首次抵达一个位置需要加上当前位置的数值，要求是全程的权值必须大于等于0，问能否到达边界位置。

分析：

到达边界位置，不妨我们先假设我们是要从左边出去，到达a[0],这样子的话a[1]-a[k]就是我们必须走的路，那么右边a[k+1]-a[n]的这些元素我们可以把他们当作是一个补给站一样的东西，小贪一下，每一次往左走之前我们先看一下右边可不可以为我们补给一点能量，这里要注意，因为是一格一格的移动，所以我们要在索取之前判断一下我们有没有这个条件，也就是说如果先碰到负数，那要先看自己有没有同等的数量来抵消（因为要保证全程都是非负的）。那么到这我们可以先对数据预处理一番：存放好每一次补给的能量以及对应的条件。判断左边之后，如果不行再试一下右边，这里需要注意的就是你的初始位置变了。

代码实现：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,k;
int a[200010];
//先假设我们是从左边出去，那么右边就当作是一个补给站一样的东西，补给能量
//存放好每次可以补给的能量并且要存放好补给的要求，也就是说有一些是小于0的，需要求一下最小值
//这样才能保证每一次补给都是合法的
//如果说左边不行再考虑右边，把左边当作是一个补给站
//最后要是还不行就是不行了
//否则输出yes，要么输出no 
bool solve()
{
	ll sum=0,mn=0;
	vector<ll>next;//存放可以补给的量以及条件
	vector<ll>need;
	for(int i=k+1;i<=n;i++)
	{
		sum+=a[i];
		mn=min(mn,sum);
		if(sum>0)
		{
			next.push_back(sum);//这两个就是存放能源补给量以及补给条件
			need.push_back(mn);
			sum=0;mn=0;//特别注意每次都需要变为0，更新
		}	
	} 
	ll cur=a[k];
	for(int i=k,j=-1;i>=1;i--)
	{
		while(j+1<next.size()&&cur+need[j+1]>=0)
		{
			j++;
			cur+=next[j];
		}
		if(cur+a[i-1]<0)	return false;
		cur+=a[i-1];
	}
	return true;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int t;
	cin>>t;
	while(t--)
	{
		cin>>n>>k;
		for(int i=1;i<=n;i++)	cin>>a[i];
		if(solve())	cout<<"YES\n";
		else//要是左边不行，那就再试一下右边
		{
			reverse(a+1,a+n+1);//翻转数组
			k=n-k+1;//之前的第k个元素需要改一下，数组翻转之后位置就变了
			if(solve())	cout<<"YES\n";
			else	cout<<"NO\n";//要是还不行，输出NO
		}
	} 
	return 0; 
}