Codeforces Round #826 (Div. 3) E. Sending a Sequence Over the Network

Problem - E - Codeforces

题意：

给定一个数组，将该数组分成连续的若干部分。每一部分的长度大小是该部分最左端或者最右端的大小-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;
}