Submission #1226884

#	Time	Username	Problem	Language	Result	Execution time	Memory
1226884		nh0902	Meteors (POI11_met)	C++20	Compilation error	0 ms	0 KiB

met

#include <bits/stdc++.h>

using namespace std;

#define int long long

const int N = 3e5 + 10;

const int M = 1e3 + 100;

int n, m, k, p[N], l[N], r[N], a[N];

vector<int> stations[N];

int cur_l[N], cur_r[N], ans[N];

vector<int> cur[N]; // cur[i] : index k that had (cur_l + cur_r) / 2 = mid;

//segment tree
int st[4 * N];

void build(int id, int l, int r) {
    st[id] = 0;
    if (l == r) return;

    int mid = (l + r) / 2;
    build(id * 2, l, mid);
    build(id * 2 + 1, mid + 1, r);
}

void update(int id, int l, int r, int u, int v, int val) {
    if (r < u || v < l) return;

    if (u <= l && r <= v) {
        st[id] += val;
        return;
    }

    int mid = (l + r) / 2;
    update(id * 2, l, mid, u, v, val);
    update(id * 2 + 1, mid + 1, r, u, v, val);
}

void update(int l, int r, int val) {
    if (l <= r) update(1, 1, m, l, r, val);
    else {
        update(1, 1, m, l, m, val);
        update(1, 1, m, 1, r, val);
    }
}

int get(int id, int l, int r, int p) {
    if (l == r) return st[id];

    int mid = (l + r) / 2;

    if (p <= mid) return st[id] + get(id * 2, l, mid, p);
    else return st[id] + get(id * 2 + 1, mid + 1, r, p);
}

int get(int i) {
    return get(1, 1, m, i);
}

void prep() {
    cin >> n >> m;
    int x;
    for (int i = 1; i <= m; i ++) {
        cin >> x;
        stations[x].push_back(i);
    }
    for (int i = 1; i <= n; i ++) {
        cin >> p[i];
    }
    cin >> k;
    for (int i = 1; i <= k; i ++) {
        cin >> l[i] >> r[i] >> a[i];
    }

    for (int i = 1; i <= n; i ++) {
        cur_l[i] = 1;
        cur_r[i] = k + 1;
    }
}

void process() {
    build(1, 1, m);
    
    for (int i = 1; i <= k; i ++) cur[i].clear();
    for (int i = 1; i <= n; i ++) {
      if (cur_l[i] < cur_r[i]) cur[(cur_l[i] + cur_r[i]) / 2].push_back(i);

    for (int i = 1; i <= k; i ++) {
        update(l[i], r[i], a[i]);

        for (int j : cur[i]) {
            int cnt = 0;
            for (int s : stations[j]) cnt += get(s);
            if (cnt >= p[j]) cur_r[j] = i;
            else cur_l[j] = i + 1;
        }
    }
}

void solve() {

    for (int i = 1; i <= log2(k) + 3; i ++) process();

    for (int i = 1; i <= n; i ++) {
        if (cur_l[i] <= k) cout << cur_l[i] << "\n";
        else cout << "NIE\n";
    }
}

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    prep();
    solve();

    return 0;
}

Compilation message (stderr)

met.cpp: In function 'void process()':
met.cpp:105:14: error: a function-definition is not allowed here before '{' token
  105 | void solve() {
      |              ^
met.cpp:115:12: warning: empty parentheses were disambiguated as a function declaration [-Wvexing-parse]
  115 | signed main() {
      |            ^~
met.cpp:115:12: note: remove parentheses to default-initialize a variable
  115 | signed main() {
      |            ^~
      |            --
met.cpp:115:12: note: or replace parentheses with braces to value-initialize a variable
met.cpp:115:15: error: a function-definition is not allowed here before '{' token
  115 | signed main() {
      |               ^
met.cpp:123:2: error: expected '}' at end of input
  123 | }
      |  ^
met.cpp:86:16: note: to match this '{'
   86 | void process() {
      |                ^