题意:
给定一个数组,将该数组分成连续的若干部分。每一部分的长度大小是该部分最左端或者最右端的大小-1,求能否将这个数组完全分成合法的若干部分。
思路:
考虑dp
状态设计:设dp[i]表示到第i个位置是否能合法
状态转移:
枚举决策,发现这次遍历到的a[i]可以作为左侧值也可以作为右侧值,如果作为左侧值,那么dp[i+a[i]]由dp[i-1]转移过来,如果作为右侧值,那么dp[i]由dp[i-a[i]-1]转移过来
Code:
#include <bits/stdc++.h>
using namespace std;
const int mxn=2e5+10;
#define int long long
int n;
int a[mxn],dp[mxn];
void solve(){
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i],dp[i]=0;
dp[0]=1;
for(int i=1;i<=n;i++){
if(i+a[i]<=n) dp[i+a[i]]|=dp[i-1];
if(i-a[i]>=1) dp[i]|=dp[i-a[i]-1];
}
if(dp[n]) cout<<"YES"<<'\n';
else cout<<"NO"<<'\n';
}
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int T=1;
cin>>T;
while(T--)solve();
return 0;
}