D. Slime Escape #822 div2

Problem - D - Codeforces

思路和代码来自cup大佬QAQ

题意是给你一个序列，有正有负，每个数相当于只能使用一次，给你开始位置，问你能不能在路途中没有变成的负的情况下，到达a0或者an+1

这题看似是个大模拟，其实真的是个大模拟，但是我觉得他的写法和思路非常好，借鉴一下。模拟部分可以略过~

既然让你到达两个边界，那就一个劲的往两边走吧

假设往a0的方向走，（往an+1的地方走只需要反转一下序列就好，走法与到a0没区别）

因为相当于每个数只能走到一次，所以可以一直往左边走，保证了不重复（我原来想的思路里面有重复了，所以不太好评价）。然后基于了一种贪心的思路，在血量充足的条件下，先增加自己的血量，增加的血量来自于后面。当不可以再增加的时候，就开始往a0的方向去，然后每一次都这样，中间加个判断就好了，如果血量小于0，就false，否则到最后就是true

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#define IOS ios::sync_with_stdio(false), cin.tie(0);
#include<iostream>
#include<map>
#include<set> 
#include<cstdio>
#include<cstring>
#include<vector>
#include<stack>
#include<algorithm>
#include<cmath>
#include<queue>
#include<deque>
using namespace std;
#define int long long
typedef long long LL;
typedef pair<int,int> PAII;
const int N=2e6+10,M=5005,INF=1e18,mod=1e9+7;
int a[N];
vector<int> add,need;
int n,k;
bool check()
{
	int sum=0,mi=0;
	for(int i=k+1;i<=n;i++)
	{
		sum+=a[i];
		mi=min(mi,sum);
		if(sum>0)
		{
			add.push_back(sum);
			need.push_back(mi);
			sum=0,mi=0;
		}
	}
	int now=a[k];
	for(int i=k,j=-1;i>=1;i--)//在每个地方就要看看是不是够补给 
	{
		while(j+1<add.size() && now+need[j+1]>=0)
		{
			j++;
			now+=add[j];
		}
		if(now+a[i-1]<0) return false;
		now+=a[i-1]; 
	}
	return true;
}
signed main(){
    //IOS;
    int T;
    //T=1;
    cin>>T;
    while(T--)
    {	
    	add.clear();
    	need.clear();
    	cin>>n>>k;
    	for(int i=1;i<=n;i++) cin>>a[i];
    	if(check())
    	{
			cout<<"YES\n";
			continue;
		}
		reverse(a+1,a+n+1);
		k=n-k+1;
		add.clear();
		need.clear();
		if(check()) cout<<"YES\n";
		else cout<<"NO\n";
    }
    return 0;
} 
/*


 
*/