Submission #447738

#	Time	Username	Problem	Language	Result	Execution time	Memory
447738		blue	Fortune Telling 2 (JOI14_fortune_telling2)	C++17	Compilation error	0 ms	0 KiB

fortune_telling2

#include <iostream>
#include <vector>
using namespace std;

const int maxK = 200'000;
const int logN = 18;
const int INF = 1'000'000'000;

int N, K;
vector<int> T(1+maxK);

struct segtree
{
    int l;
    int r;
    vector<int> v;

    segtree* left = NULL;
    segtree* right = NULL;

    segtree()
    {
        ;
    }

    segtree(int L, int R)
    {
        l = L;
        r = R;
        if(l == r)
        {
            v = vector<int>{T[l]};
        }
        else
        {
            int m = (l+r)/2;
            left = new segtree(l, m);
            right = new segtree(m+1, r);

            for(int x: left->v) v.push_back(x);
            for(int y: right->v) v.push_back(y);
            sort(v.begin(), v.end());
        }
    }

    int query(int L, int R, int X, int Y)
    {
        if(X > Y) return 0;

        if(R < l || r < L) return 0;
        else if(L <= l && r <= R)
        {
            int lo = -1;
            for(int bit = logN; bit >= 0; bit--)
            {
                if(lo + (1 << bit) >= v.size()) continue;
                if(v[lo + (1 << bit)] >= X) continue;
                lo += (1 << bit);
            }
            lo++;
            if(v[lo] < X || Y < v[lo]) return 0;

            int hi = -1;
            for(int bit = logN; bit >= 0; bit--)
            {
                if(hi + (1 << bit) >= v.size()) continue;
                if(v[hi + (1 << bit)] > Y) continue;
                hi += (1 << bit);
            }

            return hi-lo+1;
        }
        else return left->query(L, R, X, Y) + right->query(L, R, X, Y);
    }

    int get_last(int A, int B)
    {
        if(l == r) return l;
        else if(right->query(1, N, A, B) >= 1)
            return right->get_last(A, B);
        else
            return left->get_last(A, B);
    }
};

int main()
{
    cin >> N >> K;

    int A[N+1], B[N+1];
    for(int i = 1; i <= N; i++) cin >> A[i] >> B[i];

    for(int j = 1; j <= K; j++) cin >> T[j];

    segtree S(0, N);

    long long res = 0;

    for(int i = 1; i <= N; i++)
    {
        if(A[i] == B[i])
        {
            res += A[i];
            continue;
        }


        int last_pos = S.get_last(min(A[i], B[i]),  max(A[i], B[i])-1);

        if(last_pos == 0 && A[i] < B[i])
        {
            int Q = S.query(1, N, B[i], INF);

            if(Q % 2 == 0) res += A[i];
            else res += B[i];
        }
        else
        {
            int Q = S.query(last_pos + 1, N, max(A[i], B[i]), INF);

            if(Q % 2 == 0) res += max(A[i], B[i]);
            else res += min(A[i], B[i]);
        }
    }

    cout << res << '\n';
}

Compilation message (stderr)

fortune_telling2.cpp: In constructor 'segtree::segtree(int, int)':
fortune_telling2.cpp:42:13: error: 'sort' was not declared in this scope; did you mean 'qsort'?
   42 |             sort(v.begin(), v.end());
      |             ^~~~
      |             qsort
fortune_telling2.cpp: In member function 'int segtree::query(int, int, int, int)':
fortune_telling2.cpp:56:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   56 |                 if(lo + (1 << bit) >= v.size()) continue;
      |                    ~~~~~~~~~~~~~~~~^~~~~~~~~~~
fortune_telling2.cpp:66:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   66 |                 if(hi + (1 << bit) >= v.size()) continue;
      |                    ~~~~~~~~~~~~~~~~^~~~~~~~~~~