题意是给你一串序列,把他分段,分段的长度在这一段的左边或者右边。给你这样的一个序列,能不能还原成这样的每一段
这个题乍一看感觉可能用dfs写,但其实很经典的dp:每个状态是可以传递到下一个状态的
对于一个数,他很有可能是一段的长度,也有可能不是,那就以这个为分界好了
所以令f[i]为到第i个的时候是否合法
那怎么样看看是不是合法:就从上一个状态传递下来了
是不是合法,也就是他是不是某一段的值,所以你只要看看上一段的值是不是合法就可以了
比如现在的值是i,上一段是i - b[i] - 1,就是f[i - b[i] - 1]如果合法,那此时的f[i]也是合法的
注意:注意一下边界,并且f[0] = 1
那如果此时的i是合法的,那下一段的i + b[i] + 1 ,此时就是f[i + b[i] + 1] 也是合法的
这样遍历一遍,就可以了
下面详见代码:
#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=5050,INF=1e18,mod=998244353;
int f[N], a[N];
signed main(){
//IOS;
int T;
//T=1;
cin>>T;
while(T--)
{
int n;
cin >> n;
for(int i = 1; i <= n; i ++)
{
cin >> a[i];
f[i] = 0;
}
f[0] = 1;
for(int i = 0; i <= n; i ++)
{
if(i - a[i] - 1 >= 0 && f[i - a[i] - 1]) f[i] = 1;
if(!f[i]) continue;
if(i + a[i + 1] + 1 <= n) f[i + a[i + 1] + 1] = 1;
}
if(f[n]) cout << "YES\n";
else cout << "NO\n";
}
return 0;
}
/*
*/