Submission #953187

#TimeUsernameProblemLanguageResultExecution timeMemory
953187underwaterkillerwhaleFortune Telling 2 (JOI14_fortune_telling2)C++17
100 / 100
674 ms77252 KiB
#include <bits/stdc++.h>
#define se          second
#define fs          first
#define pb          push_back
#define ll          long long
#define ii          pair<ll,ll>
#define ld          long double
#define SZ(v)       (int)v.size()
#define ALL(v)      v.begin(), v.end()
#define bit(msk, i) ((msk >> i) & 1)
#define iter(id, v) for (auto id : v)
#define rep(i,m,n)  for(int i=(m); i<=(n); i++)
#define reb(i,m,n)  for(int i=(m); i>=(n); i--)

using namespace std;

mt19937_64 rd(chrono :: steady_clock :: now().time_since_epoch().count());
ll Rand(ll l, ll r) { return uniform_int_distribution<ll> (l, r)(rd); }


const int N = 2e5 + 7;
const int Mod = 1e9 + 7;
const int szBL = 500;
const int INF = 2e9 + 7;
const int BASE = 137;

struct Arr_Data {
    int a, b;
    bool isswapped;
    Arr_Data () : a(0), b(0), isswapped(0) {};
};

///input:
int n, Q;
Arr_Data arr[N];
int qr[N];

///compress:
int num_Val;
vector<int> nen;

int CPR (int X) {
    return lower_bound(ALL(nen), X) - nen.begin() + 1;
}

void compress () {
    nen = {0, INF};
    rep (i, 1, n) nen.push_back(arr[i].a), nen.push_back(arr[i].b);
    rep (i, 1, Q) nen.push_back(qr[i]);
    sort (ALL(nen));
    nen.resize(num_Val = unique(ALL(nen)) - nen.begin());
    rep (i, 1, n) {
        arr[i].a = CPR(arr[i].a);
        arr[i].b = CPR(arr[i].b);
    }
    rep (i, 1, Q) qr[i] = CPR(qr[i]);
}

///data:
vector<vector<int>> queries, events;

class Segment_Tree_helper1 {
private:
    int Range;
    int st[3][N << 2];
    void update (int id, int l, int r, int pos, int val) {
        if (l > pos || r < pos) return;
        if (l == r) {
            st[1][id] = st[0][id] = st[2][id] = val;
            return;
        }
        int mid = l + r >> 1;
        update (id << 1, l, mid, pos, val);
        update (id << 1 | 1, mid + 1, r, pos, val);
        st[0][id] = min(st[0][id << 1], st[0][id << 1 | 1]);
        st[1][id] = max(st[1][id << 1], st[1][id << 1 | 1]);
        st[2][id] = st[2][id << 1] + st[2][id << 1 | 1];
    }
    int get (int id, int l, int r, int u, int v, int typ) {
        if (l > v || r < u) {
            if (typ == 0) return INF;
            else return 0;
        }
        if (l >= u && r <= v) return st[typ][id];
        int mid = l + r >> 1;
        if (typ == 1) return max(get (id << 1, l, mid, u, v, typ), get (id << 1 | 1, mid + 1, r, u, v, typ));
        else if (typ == 0) return min(get (id << 1, l, mid, u, v, typ), get (id << 1 | 1, mid + 1, r, u, v, typ));
        else return get (id << 1, l, mid, u, v, typ) + get (id << 1 | 1, mid + 1, r, u, v, typ);
    }
    int walk_Left (int id, int l, int r, int bound) { ///walk to Find L when the upper_bound is bound
        if (l > bound) return -1;
        if (st[1][id] == 0) return -1;
        if (l == r) return l;
        int mid = l + r >> 1;
        int res = -1;
        if (mid + 1 <= bound) res = walk_Left (id << 1 | 1 , mid + 1, r, bound);
        if (res == -1) res = walk_Left (id << 1, l, mid, bound);
        return res;
    }
    int walk_Right (int id, int l, int r, int bound) {  ///walk to Find R when the lower_bound is bound
        if (r < bound) return -1;
        if (st[1][id] == 0) return -1;
        if (l == r) return l;
        int mid = l + r >> 1;
        int res = -1;
        if (mid >= bound) res = walk_Right (id << 1, l, mid, bound);
        if (res == -1) res = walk_Right (id << 1 | 1, mid + 1, r, bound);
        return res;
    }
    int search_rightmost_pole (int id, int l, int r, int bound) {
        if (st[1][id] < bound) return -1;
        if (l == r) return l;
        int mid = l + r >> 1;
        int res = -1;
        if (st[1][id << 1 | 1] >= bound) res = search_rightmost_pole (id << 1 | 1, mid + 1, r, bound);
        if (res == -1) res = search_rightmost_pole (id << 1, l, mid, bound);
        return res;
    }

public:
    void init(int n ) {
        Range = n;
        rep (i, 0, (Range + 1) << 2) st[0][i] = INF, st[1][i] = 0, st[0][i] = 0;
    }
    void update (int pos, int val) { update (1, 0, Range, pos, val); }
    int getMx (int L, int R) { return get (1, 0, Range, L, R, 1); }
    int getMn (int L, int R) { return get (1, 0, Range, L, R, 0);}
    int Sum (int L, int R) { return get (1, 0, Range, L, R, 2); }
    int walk_Left (int bound){ return walk_Left (1, 0, Range, bound); }
    int walk_Right (int bound) { return walk_Right (1, 0, Range, bound); }
    int search_rightmost_pole (int bound) { return search_rightmost_pole (1, 0, Range, bound); }

}ST, Partition, values_Range;

void solution() {
    cin >> n >> Q;
    rep (i, 1, n) {
        cin >> arr[i].a >> arr[i].b;
    }
    rep (i, 1, Q) {
        cin >> qr[i];
    }
    compress();

    events.resize(num_Val + 1, vector<int> (0)), queries.resize(num_Val + 1, vector<int> (0));
    rep (i, 1, n) {
        if (arr[i].a > arr[i].b) {
            swap(arr[i].a, arr[i].b);
            arr[i].isswapped = 1;
        }
        events[arr[i].b].push_back(i);
    }
    rep (i, 1, Q) queries[qr[i]].push_back(i);

    ST.init(Q), Partition.init(Q), values_Range.init(Q + 1);
    rep (i, 1, Q) ST.update (i, qr[i]);
    values_Range.update (Q + 1, ST.getMx(1, Q));
    vector<int> final_arr(n + 1, 0);
    reb (X, num_Val, 1) {
        iter (&id, queries[X]) {
            int boundL = Partition.walk_Left(id - 1), boundR = Partition.walk_Right(id + 1);
            if (boundL == -1) boundL = 0;
            if (boundR == -1) boundR = Q + 1;
            Partition.update (id, 1);
            values_Range.update (id, ST.getMx(boundL + 1, id - 1));
            values_Range.update (boundR, ST.getMx(id + 1, boundR - 1));
        }
        iter (&id, events[X]) {
            int A = arr[id].a, B = arr[id].b, swapped = arr[id].isswapped;
            int rightmost = values_Range.search_rightmost_pole(A);
            bool ok = rightmost == -1 ? swapped : 0;
            if (rightmost == -1) rightmost = 0;
            int parity = Partition.Sum (rightmost + 1, Q) & 1;
            if (parity ^ ok & 1) final_arr[id] = nen[B - 1];
            else {
                if (values_Range.getMx(Q + 1, Q + 1) >= A) final_arr[id] = nen[B - 1];
                else final_arr[id] = nen[A - 1];
            }
        }
    }
    ll res = 0;
    rep (i, 1, n) {
        res += final_arr[i];
    }
    cout<<res <<"\n";
}

#define file(name) freopen(name".inp", "r", stdin); freopen(name".out", "w", stdout);

int main() {
//    file("c");
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    ll num_Test = 1;
//    cin >> num_Test;
    while(num_Test--)
        solution();
}
/*


*/

Compilation message (stderr)

fortune_telling2.cpp: In member function 'void Segment_Tree_helper1::update(int, int, int, int, int)':
fortune_telling2.cpp:72:21: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   72 |         int mid = l + r >> 1;
      |                   ~~^~~
fortune_telling2.cpp: In member function 'int Segment_Tree_helper1::get(int, int, int, int, int, int)':
fortune_telling2.cpp:85:21: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   85 |         int mid = l + r >> 1;
      |                   ~~^~~
fortune_telling2.cpp: In member function 'int Segment_Tree_helper1::walk_Left(int, int, int, int)':
fortune_telling2.cpp:94:21: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   94 |         int mid = l + r >> 1;
      |                   ~~^~~
fortune_telling2.cpp: In member function 'int Segment_Tree_helper1::walk_Right(int, int, int, int)':
fortune_telling2.cpp:104:21: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
  104 |         int mid = l + r >> 1;
      |                   ~~^~~
fortune_telling2.cpp: In member function 'int Segment_Tree_helper1::search_rightmost_pole(int, int, int, int)':
fortune_telling2.cpp:113:21: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
  113 |         int mid = l + r >> 1;
      |                   ~~^~~
fortune_telling2.cpp: In function 'void solution()':
fortune_telling2.cpp:174:29: warning: suggest parentheses around arithmetic in operand of '^' [-Wparentheses]
  174 |             if (parity ^ ok & 1) final_arr[id] = nen[B - 1];
      |                          ~~~^~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...