E. Sending a Sequence Over the Network #826 div3

慎独666

已于 2022-10-12 15:42:49 修改

阅读量467

点赞数 2

分类专栏： acm 文章标签：算法

于 2022-10-12 15:42:47 首次发布

本文链接：https://blog.csdn.net/m0_63305704/article/details/127283831

版权

acm 专栏收录该内容

98 篇文章 0 订阅

订阅专栏

Problem - E - Codeforces

题意是给你一串序列，把他分段，分段的长度在这一段的左边或者右边。给你这样的一个序列，能不能还原成这样的每一段

这个题乍一看感觉可能用dfs写，但其实很经典的dp：每个状态是可以传递到下一个状态的

对于一个数，他很有可能是一段的长度，也有可能不是，那就以这个为分界好了

所以令f[i]为到第i个的时候是否合法

那怎么样看看是不是合法：就从上一个状态传递下来了

是不是合法，也就是他是不是某一段的值，所以你只要看看上一段的值是不是合法就可以了

比如现在的值是i，上一段是i - b[i] - 1，就是f[i - b[i] - 1]如果合法，那此时的f[i]也是合法的

注意：注意一下边界，并且f[0] = 1

那如果此时的i是合法的，那下一段的i + b[i] + 1 ，此时就是f[i + b[i] + 1] 也是合法的

这样遍历一遍，就可以了

下面详见代码：

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#define IOS ios::sync_with_stdio(false), cin.tie(0);
#include<iostream>
#include<map>
#include<set> 
#include<cstdio>
#include<cstring>
#include<vector>
#include<stack>
#include<algorithm>
#include<cmath>
#include<queue>
#include<deque>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> PAII;
const int N=2e6+10,M=5050,INF=1e18,mod=998244353;
int f[N], a[N];
signed main(){
    //IOS;
    int T;
    //T=1;
    cin>>T;
    while(T--)
    {
		int n;
		cin >> n;
		for(int i = 1; i <= n; i ++)
		{
			cin >> a[i];
			f[i] = 0;
		}
		f[0] = 1;
		for(int i = 0; i <= n; i ++)
		{
			if(i - a[i] - 1 >= 0 && f[i - a[i] - 1]) f[i] = 1;
			if(!f[i]) continue;
			if(i + a[i + 1] + 1 <= n) f[i + a[i + 1] + 1] = 1; 
		}
		if(f[n]) cout << "YES\n";
		else cout << "NO\n";
    }
    return 0;
} 
/* 
 
 
*/