Submission #19559659

---

Source Code Expand

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
#define FOR(i, a, b) for(int i = (a); i < (b); ++i)
#define REP(i, n) for(int i = 0; i < (n); ++i)
#define REPr(i, n) for(int i = (n)-1; i >= 0; --i)
#define FORq(i, m, n) for(int i = (m); i <= (n); ++i)
#define FORqr(i, m, n) for(int i = (n); i >= (m); --i)
#define MP make_pair
#define SIN(x, S) (S.count(x) != 0)
#define M0(x) memset(x, 0, sizeof(x))
#define FILL(x, y) memset(x, y, sizeof(x))
#define MM(x) memset(x, -1, sizeof(x))
#define ALL(x) (x).begin(), (x).end()
#define DB(x) cerr << #x << " = " << x << endl
#define DB2(x, y)                                                              \
    cerr << "(" << #x << ", " << #y << ") = (" << x << ", " << y << ")\n";
#define DB3(x, y, z)                                                           \
    cerr << "(" << #x << ", " << #y << ", " << #z << ") = (" << x << ", " << y << ", " << z << ")\n";
typedef pair<int, int> PII;
typedef pair<long long, long long> PLL;
typedef vector<int> VI;
typedef vector<VI> VVI;
typedef vector<long long> VL;
typedef long long ll;
typedef long long integer;
const long long MOD = 1e9 + 7;


///////////////////////////////////////////////
// for template
template <class T> inline bool chmin(T &a, T b) {
    if(a > b) {
        a = b;
        return true;
    }
    return false;
}
template <class T> inline bool chmax(T &a, T b) {
    if(a < b) {
        a = b;
        return true;
    }
    return false;
}
///////////////////////////////////////////////
/// 🍈( '-' 🍈 |AC|
ll N;
string T;

ll solve(ll N, string T){
    ll Z = 0;
    for (auto t: T){
        if (t == '0') Z++;
    }

    REP(i,N - 2){
        if ((T[i] == '1') and (T[i + 1] == '1') and (T[i + 2] == '1')){
            return 0;
        }
    }

    if (Z == 0){
        if (N == 1) return 2 * 1e10;
        if (N == 2) return 1e10;
    }else if (Z == 1){
        ll x = 0;
        ll y = 0;
        bool z = false;
        for (auto t : T){
            if (t == '0') z = true;
            if (t == '1'){
                if (z) y++;
                else x++;
            }
        }
        if (y == 0) return 1e10;
        else return 1e10 - 1;
    }else{
        ll x = 0;
        ll y = 0;
        int m = 0;
        REP(i, N){
            char c = T[i];
            if (c == '1') x++;
            else break;
        }
        
        REPr(i, N){
            char c = T[i];
            if (c == '1') y++;
            else break;
        }
        
        //DB2(x,y);
        FORq(i,x,N - 1 - y){
            int j = i - x;
            if (j % 3 == 0){
                if (T[i] != '0') return 0;
            }else{
                if (T[i] != '1') return 0;
            }
        }

        return 1e10 - Z + 1 - (y != 0);
    }

}

int main() {
    cin >> N;
    cin >> T;
    cout << solve(N, T) << endl;
}

Submission Info

Compile Error

./Main.cpp: In function ‘ll solve(ll, std::string)’:
./Main.cpp:83:13: warning: unused variable ‘m’ [-Wunused-variable]
   83 |         int m = 0;
      |             ^
./Main.cpp:109:1: warning: control reaches end of non-void function [-Wreturn-type]
  109 | }
      | ^

Judge Result