제출 #501671

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
501671		tengiz05	Crossing (JOI21_crossing)	C++17	3 / 100	7060 ms	43760 KiB

Main

#include <bits/stdc++.h>

using i64 = long long;

int pre[9][3][200005];

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    
    int n;
    std::cin >> n;
    
    auto in = [&]() {
        std::string a;
        std::cin >> a;
        std::vector<int> b(n);
        for (int i = 0; i < n; i++) {
            b[i] = std::string("JOI").find(a[i]);
        }
        return b;
    };
    
    std::vector<int> a = in(), b = in(), c = in();
    
    auto merge = [&](std::vector<int> a, std::vector<int> b) {
        auto c = a;
        for (int i = 0; i < n; i++) {
            if (a[i] != b[i]) {
                c[i] = 0 ^ 1 ^ 2 ^ a[i] ^ b[i];
            }
        }
        return c;
    };
    
    std::vector<std::vector<int>> s{a, b, c, merge(a, b), merge(a, c), merge(b, c), merge(merge(a, c), b), merge(merge(a, b), c), merge(merge(b, c), a)};
    
    int q;
    std::cin >> q;
    std::vector<int> t = in();
    
    for (int k = 0; k < 9; k++) {
        for (int i = 0; i < 3; i++) {
            for (int j = 0; j < n; j++) {
                pre[k][i][j + 1] = pre[k][i][j] + (s[k][j] == i);
            }
        }
    }
    
    int bad[9] {};
    for (int k = 0; k < 9; k++)
        for (int i = 0; i < n; i++)
            bad[k] += t[i] != s[k][i];
    
    std::set<std::pair<int, int>> p{{-1, -1}, {n, -1}};
    for (int i = 0; i < n; i++)
        p.emplace(i, t[i]);
    
    int ok = 0;
    for (int k = 0; k < 9; k++) {
        ok += bad[k] == 0;
    }
    if (ok) {
        std::cout << "Yes\n";
    } else {
        std::cout << "No\n";
    }
    
    auto rem = [&](int l, int r, int c) {
        for (int k = 0; k < 9; k++) {
            bad[k] -= (pre[k][c][r] - pre[k][c][l] != r - l);
        }
    };
    auto add = [&](int l, int r, int c) {
        for (int k = 0; k < 9; k++) {
            bad[k] += (pre[k][c][r] - pre[k][c][l] != r - l);
        }
    };
    
    while (q--) {
        int l, r;
        char ch;
        std::cin >> l >> r >> ch;
        l--;
        
        int x = std::string("JOI").find(ch);
        
        for (int i = l; i < r; i++) {
            t[i] = x;
        }
        
        if (s[0] == t) {
            std::cout << "Yes\n";
        } else {
            std::cout << "No\n";
        }
        
        // auto it = p.lower_bound({l, -1});
        // if (it->first > l) {
            // int L = std::prev(it)->first;
            // int R = it->first;
            // int c = std::prev(it)->second;
            // rem(L, R, c);
            // add(L, l, c);
            // add(l, R, c);
            // p.emplace(l, c);
        // }
        
        // it = p.lower_bound({r, -1});
        // if (it->first > r) {
            // int L = std::prev(it)->first;
            // int R = it->first;
            // int c = std::prev(it)->second;
            // rem(L, R, c);
            // add(L, r, c);
            // add(r, R, c);
            // p.emplace(r, c);
        // }
        
        // assert(p.lower_bound({l, -1})->first == l);
        // while (true) {
            // it = p.lower_bound({l, -1});
            // if (it->first < r) {
                // rem(it->first, std::next(it)->first, it->second);
                // p.erase(it);
            // } else {
                // break;
            // }
        // }
        
        // add(l, r, x);
        // p.emplace(l, x);
        
        // int ok = 0;
        // for (int k = 0; k < 9; k++) {
            // ok += bad[k] == 0;
        // }
        // if (ok) {
            // std::cout << "Yes\n";
        // } else {
            // std::cout << "No\n";
        // }
    }
    
    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

Main.cpp: In function 'int main()':
Main.cpp:69:10: warning: variable 'rem' set but not used [-Wunused-but-set-variable]
   69 |     auto rem = [&](int l, int r, int c) {
      |          ^~~
Main.cpp:74:10: warning: variable 'add' set but not used [-Wunused-but-set-variable]
   74 |     auto add = [&](int l, int r, int c) {
      |          ^~~

• Subtask 1: 3 / 3
• Subtask 2: 0 / 23
• Subtask 3: 0 / 23
• Subtask 4: 0 / 51