Submission #873402

#	Time	Username	Problem	Language	Result	Execution time	Memory
873402		garam1732	Snake Escaping (JOI18_snake_escaping)	C++14	75 / 100	2098 ms	48636 KiB

snake_escaping

#include <bits/stdc++.h>
using namespace std;

#define ff first
#define ss second
#define bl " "
#define endl "\n"
#define all(v) v.begin(), v.end()
#define comp(v) v.erase(unique(all(v)), v.end())
typedef long long ll;
typedef pair<int, int> pi;
typedef pair<pi, int> pii;
typedef pair<ll, ll> pll;

const int MAXN = 1000100/5;
const int MOD = 1e9+7;

struct Query {
    string s; int x;
}qry[MAXN];

queue<Query> qq[3];
vector<int> v;
int ans[MAXN];

void solve(int s, int e, int l, int r, int x) {
    if(s > e) return;
    if(l == r) {
        for(int i = s; i <= e; i++) ans[qry[i].x] = v[l];
        return;
    }

    int a = s, mid = l+r>>1;
    for(a = s; a <= e; a++) {
        if(qry[a].s[x] != '0') break;
    }

    solve(s, a-1, l, mid, x+1);

    s = a;
    for(; a <= e; a++) {
        if(qry[a].s[x] != '1') break;
    }

    solve(s, a-1, mid+1, r, x+1);

    s = a;
    for(int i = l, j = mid+1; j <= r; i++, j++) v[i] += v[j];
    solve(s, e, l, mid, x+1);
    for(int i = l, j = mid+1; j <= r; i++, j++) v[i] -= v[j];
}

int main() {
    ios_base::sync_with_stdio(0); cin.tie(0);

    int l, q; cin >> l >> q; char x;
    for(int i = 0; i < (1<<l); i++) {
        cin >> x; v.push_back(x-'0');
    }

    for(int i = 0; i < 5; i++) {
        int s = q*i/5+1, e = q*(i+1)/5;

        for(int j = s; j <= e; j++) {
            cin >> qry[j-s].s; qry[j-s].x = j-s;
        }

        //sorting
        for(int j = l-1; j >= 0; j--) {
            for(int k = 0; k <= e-s; k++) {
                if(qry[k].s[j] == '0') qq[0].push(qry[k]);
                else if(qry[k].s[j] == '1') qq[1].push(qry[k]);
                else qq[2].push(qry[k]);
            }

            int idx = 0;
            for(int k = 0; k < 3; k++) {
                while(qq[k].size()) {
                    qry[idx++] = qq[k].front();
                    qq[k].pop();
                }
            }
        }

        solve(0, e-s, 0, (1<<l)-1, 0);

        for(int j = 0; j <= e-s; j++) cout << ans[j] << endl;
    }
}

Compilation message (stderr)

snake_escaping.cpp: In function 'void solve(int, int, int, int, int)':
snake_escaping.cpp:33:23: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   33 |     int a = s, mid = l+r>>1;
      |                      ~^~

• Subtask 1: 5 / 5
• Subtask 2: 7 / 7
• Subtask 3: 10 / 10
• Subtask 4: 53 / 53
• Subtask 5: 0 / 25