CF1741E Sending a Sequence Over the Network 动态规划

CF1741E Sending a Sequence Over the Network

题意

给定一个数组，判断是否能够满足下列条件：

• 1.分块（连续，完全覆盖）
• 2.对于每块，序列长度被写在左边或右边

思路

此题为动态规划题。用dp[i]代表以i结尾的前缀方案是否合法，合法为1，反之为0。
状态转移：

• 如果以第i个数结尾，这个数为a[i],先考虑它是左侧a[i]个数的长度，并且dp[i-a[i]-1]即这a[i]+1个数之前的所有数必须要已经是合法方案了，那么以第i个数结尾的这个数才可以是一个合法方案。
• 考虑右侧a[i]个数的长度，那么说明dp[i-1]需要已经是一个合法方案了，这样右侧的a[i]+1个数才是合法的，即第a[i]+1个数之前都是合法。

代码

#include <bits/stdc++.h>

#define ll long long

using namespace std;
const int maxn=0x3f3f3f3f;

int dp[200010];
int a[200010];

void solve()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;++i)
        cin>>a[i];
    memset(dp,0,sizeof dp);
    dp[0]=1;
    for(int i=1;i<=n;++i)
    {
        if(i-a[i]>=1&&dp[i-a[i]-1])
            dp[i]=1;
        if(i+a[i]<=n&&dp[i-1])
            dp[i+a[i]]=1;
    }
    cout<<(dp[n]?"YES":"NO")<<endl;
}

int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int t;
    cin>>t;
    while (t--)
    {
        solve();
    }
    return 0;
}