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;
}