CF1741E 每日一题 Sending a Sequence Over the Network

题目链接：Problem - E - Codeforces

 input：

7
9
1 1 2 3 1 3 2 2 3
5
12 1 2 7 5
6
5 7 8 9 10 3
4
4 8 6 2
2
3 1
10
4 6 2 1 9 4 9 3 4 2
1
1

output：

YES
YES
YES
NO
YES
YES
NO

题意：

给定一个长度为 n 的序列 b，询问该序列否按以下要求拆分成若干个段：

每个段包含至少一个数，并且在该段的左边或右边，有一个数来表示该段的长度（这个数等于该段长度，具体见题目例子）。

思路：

对于序列里的每一个数，有三种考虑情况：

1.该数为某段中的数，显然这个数的大小没有意义。

2.该数将在某段的左边，并表示其长度，假设该数为 k ，位置为 i ，那么a[i] ~ a[i + k]为一段，那么第 i + k + 1 位数又可以在下一段的左边，表示其长度，当然，前提是第 i 位可以作为表示数，所以转移方程为： dp[i + a[i] + 1] |= dp[i]

3.该数将在某段的右边，并表示其长度，假设该数为 k ，位置为 i ，那么a[i] ~ a[i - k]为一段，那么第 i - k - 1 位数又可以在下一段的右边，表示其长度，当然，前提是第 i 位可以作为表示数，所以转移方程为： dp[i] |= dp[i - 1 - a[i - 1]

注意每个数要考虑边界条件！

Code

#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define endl "\n"
#define pb push_back
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int mod=1e9+7;
const int N=2e5+10;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3f;

int t,n;
int a[N];
bool dp[N];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>t;
	while(t--)
	{
		cin>>n;
		for(int i=1;i<=n+1;i++)
		{
			dp[i]=false;
		}
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];
		}
		dp[1]=true;
		for(int i=1;i<=n+1;i++)
		{
			if(i>1&&i-1-a[i-1]>=1)
			{
				dp[i]|=dp[i-1-a[i-1]];
			}
			if(i<=n&&i+a[i]+1<=n+1)
			{
				dp[i+a[i]+1]|=dp[i];
			}
		}
		if(dp[n+1])
			cout<<"YES"<<endl;
		else
		    cout<<"NO"<<endl;
	}
	return 0;
}