Codeforces 1462B Last Year's Substring

题意

给一个长度为$n$的字符串，问能否将其中一段子串（可以为空）删除，使得剩下的内容拼接起来成为$’2020’$。

分析

若删除空串，则原字符串要为$’2020’$。
若保留第一个字符，则第一个字符为$’2’$，倒数三个字符为$’020’$。

保留前两个、前三个、前四个字符同理。

#pragma GCC optimize(3, "Ofast", "inline")

#include <bits/stdc++.h>

#define start ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ll long long
#define ull unsigned long long
#define int ll
#define ls st<<1
#define rs st<<1|1
#define pii pair<int,int>
#define rep(z, x, y) for(int z=x;z<=y;++z)
#define repd(z, x, y) for(int z=x;z>=y;--z)
#define com bool operator<(const node &b)const
using namespace std;
mt19937 rnd(chrono::high_resolution_clock::now().time_since_epoch().count());
const int maxn = (ll) 1e6 + 5;
const int mod = 998244353;
const int inf = 0x3f3f3f3f;
int T = 1;

void solve() {
    int n;
    cin >> n;
    string s;
    cin >> s;
    if (s == "2020") {
        cout << "YES\n";
        return;
    } else {
        if (s.substr(s.size() - 4) == "2020") {
            cout << "YES\n";
            return;
        }
        if (s[0] == '2' && s.substr(s.size() - 3) == "020") {
            cout << "YES\n";
            return;
        }
        if (s[0] == '2' && s[1] == '0' && s.substr(s.size() - 2) == "20") {
            cout << "YES\n";
            return;
        }
        if (s[0] == '2' && s[1] == '0' && s[2] == '2' && s.substr(s.size() - 1) == "0") {
            cout << "YES\n";
            return;
        }
        if (s[0] == '2' && s[1] == '0' && s[2] == '2' && s[3] == '0') {
            cout << "YES\n";
            return;
        }
    }
    cout << "NO\n";
}

signed main() {
    start;
    cin >> T;
    while (T--)
        solve();
    return 0;
}