#489564 (C++17) No.924 紲星

提出ソース

結果

問題	No.924 紲星
ユーザー	jell
提出日時	2020-05-30 11:53:35
言語	C++17 (gcc 13.2.0 + boost 1.83.0)
結果	RE
実行時間	-
コード長	8,545 bytes
コンパイル時間	2,998 ms
コンパイル使用メモリ	158,632 KB
実行使用メモリ	10,596 KB
最終ジャッジ日時	2023-08-07 04:47:08
合計ジャッジ時間	9,622 ms
ジャッジサーバーID （参考情報）	judge15 / judge13

このコードへのチャレンジ（β）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	RE	-
testcase_01	RE	-
testcase_02	RE	-
testcase_03	RE	-
testcase_04	RE	-
testcase_05	RE	-
testcase_06	RE	-
testcase_07	RE	-
testcase_08	RE	-
testcase_09	RE	-
testcase_10	RE	-
testcase_11	RE	-
testcase_12	RE	-
testcase_13	RE	-
testcase_14	RE	-
testcase_15	RE	-
testcase_16	RE	-
testcase_17	RE	-
testcase_18	RE	-

権限があれば一括ダウンロードができます

コンパイルメッセージ

main.cpp:260:1: 警告: ISO C++ では型の無い ‘main’ の宣言を禁止しています [-Wreturn-type]
  260 | main()
      | ^~~~

ソースコード

raw source code

#ifdef LOCAL
    #define _GLIBCXX_DEBUG
    #define __clock__
#else
    #pragma GCC optimize("Ofast")
    // #define NDEBUG
#endif
#define __precision__ 10
#define iostream_untie true
#define debug_stream std::cerr
#include <algorithm>
#include <bitset>
#include <cassert>
#include <chrono>
#include <complex>
#include <cstring>
#include <deque>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <random>
#include <set>
#include <stack>
#include <unordered_map>
#include <unordered_set>
#define all(v) std::begin(v), std::end(v)
#define rall(v) std::rbegin(v), std::rend(v)
#define odd(n) ((n) & 1)
#define even(n) (not __odd(n))
#define __popcount(n) __builtin_popcountll(n)
#define __clz32(n) __builtin_clz(n)
#define __clz64(n) __builtin_clzll(n)
#define __ctz32(n) __builtin_ctz(n)
#define __ctz64(n) __builtin_ctzll(n)

using i32 = int_least32_t; using i64 = int_least64_t; using u32 = uint_least32_t; using u64 = uint_least64_t;
using pii = std::pair<i32, i32>; using pll = std::pair<i64, i64>;
template <class T> using heap = std::priority_queue<T>;
template <class T> using rheap = std::priority_queue<T, std::vector<T>, std::greater<T>>;
template <class K, class T> using hashmap = std::unordered_map<K, T>;
template <class T> using hashset = std::unordered_set<T>;

namespace std
{
    template <class P> void rsort(P __first, P __last) { sort(__first, __last, greater<>()); }
    template <class T> size_t hash_combine(size_t seed, T const &key) { return seed ^ (hash<T>()(key) + 0x9e3779b9 + (seed << 6) + (seed >> 2)); }
    template <class T, class U> struct hash<pair<T, U>> { size_t operator()(pair<T, U> const &pr) const { return hash_combine(hash_combine(0, pr.first), pr.second); } };
    template <class tuple_t, size_t index = tuple_size<tuple_t>::value - 1> struct tuple_hash_calc { static size_t apply(size_t seed, tuple_t const &t) { return hash_combine(tuple_hash_calc<tuple_t, index - 1>::apply(seed, t), get<index>(t)); } };
    template <class tuple_t> struct tuple_hash_calc<tuple_t, 0> { static size_t apply(size_t seed, tuple_t const &t) { return hash_combine(seed, get<0>(t)); } };
    template <class... T> struct hash<tuple<T...>> { size_t operator()(tuple<T...> const &t) const { return tuple_hash_calc<tuple<T...>>::apply(0, t); } };
    template <class T, class U> istream &operator>>(std::istream &s, pair<T, U> &p) { return s >> p.first >> p.second; }
    template <class T, class U> ostream &operator<<(std::ostream &s, const pair<T, U> &p) { return s << p.first << " " << p.second; }
    template <class T> istream &operator>>(istream &s, vector<T> &v) { for(T &e : v) s >> e; return s; }
    template <class T> ostream &operator<<(ostream &s, const vector<T> &v)  { bool is_front = true; for(const T &e : v) { if(not is_front) s << ' '; else is_front = false; s << e; } return s; }
    template <class tuple_t, size_t index> struct tupleos { static ostream &apply(ostream &s, const tuple_t &t) { tupleos<tuple_t, index - 1>::apply(s, t); return s << " " << get<index>(t); } };
    template <class tuple_t> struct tupleos<tuple_t, 0> { static ostream &apply(ostream &s, const tuple_t &t) { return s << get<0>(t); } };
    template <class... T> ostream &operator<<(ostream &s, const tuple<T...> &t)  { return tupleos<tuple<T...>, tuple_size<tuple<T...>>::value - 1>::apply(s, t); }
    template <> ostream &operator<<(ostream &s, const tuple<> &t) { return s; }
    string revstr(string str) { reverse(str.begin(), str.end()); return str; }
} // namespace std

#ifdef LOCAL
    #define dump(...)                                                              \
        debug_stream << "[ " << __LINE__ << " : " << __FUNCTION__ << " ]\n",       \
            dump_func(#__VA_ARGS__, __VA_ARGS__)
    template <class T> void dump_func(const char *ptr, const T &x)
    {
        debug_stream << '\t';
        for(char c = *ptr; c != '\0'; c = *++ptr) if(c != ' ') debug_stream << c;
        debug_stream << " : " << x << '\n';
    }
    template <class T, class... rest_t> void dump_func(const char *ptr, const T &x, rest_t... rest)
    {
        debug_stream << '\t';
        for(char c = *ptr; c != ','; c = *++ptr) if(c != ' ') debug_stream << c;
        debug_stream << " : " << x << ",\n"; dump_func(++ptr, rest...);
    }
#else
    #define dump(...) ((void)0)
#endif
template <class P> void read_range(P __first, P __second) { for(P i = __first; i != __second; ++i) std::cin >> *i; }
template <class P> void write_range(P __first, P __second) { for(P i = __first; i != __second; std::cout << (++i == __second ? '\n' : ' ')) std::cout << *i; }

// substitue y for x if x > y.
template <class T> inline bool sbmin(T &x, const T &y) { return x > y ? x = y, true : false; }
// substitue y for x if x < y.
template <class T> inline bool sbmax(T &x, const T &y) { return x < y ? x = y, true : false; }
// binary search.
i64 bin(const std::function<bool(i64)> &pred, i64 ok, i64 ng)
{
    while(std::abs(ok - ng) > 1) { i64 mid = (ok + ng) / 2; (pred(mid) ? ok : ng) = mid; }
    return ok;
}
double bin(const std::function<bool(double)> &pred, double ok, double ng, const double eps)
{
    while(std::abs(ok - ng) > eps) { double mid = (ok + ng) / 2; (pred(mid) ? ok : ng) = mid; }
    return ok;
}
// be careful that val is type-sensitive.
template <class T, class A, size_t N> void init(A (&array)[N], const T &val) { std::fill((T *)array, (T *)(array + N), val); }
// reset all bits.
template <class A> void reset(A &array) { memset(array, 0, sizeof(array)); }


/* The main code follows. */

using namespace std;


class Mo
{
    using upd_t = std::function<void(size_t)>;
    const upd_t &add, &rem;
    std::vector<size_t> lft, rgt, ord;
    bool is_built;
    const size_t width;
    size_t nl, nr;
    std::vector<size_t>::iterator itr;

public:
    Mo(size_t n, const upd_t &_add, const upd_t &_rem)
        : add(_add), rem(_rem), is_built(), width(sqrt(n)), nl(0), nr(0)
    {}

    size_t left() const { return nl; }
    size_t right() const { return nr; }

    void insert(size_t l, size_t r)
    {
        lft.emplace_back(l), rgt.emplace_back(r);
        is_built = false;
    }

    void build()
    {
        is_built = true;
        ord.resize(lft.size());
        std::iota(ord.begin(), ord.end(), 0);
        std::sort(ord.begin(), ord.end(),
        [&](size_t x, size_t y)
        {
            if(lft[x] / width != lft[y] / width) return lft[x] < lft[y];
            return rgt[x] < rgt[y];
        });
        itr = ord.begin();
    }

    size_t process()
    {
        if(!is_built) build();
        if(itr == ord.end()) return ord.size();
        const size_t now = *itr;
        while(nl > lft[now]) add(--nl);
        while(nr < rgt[now]) add(nr++);
        while(nl < lft[now]) rem(nl++);
        while(nr > rgt[now]) rem(--nr);
        return *itr++;
    }
};

int a[1<<18];
i64 lsum,rsum;
multiset<int> lit,lar;

int ltop()
{
    assert(!lit.empty());
    return *lit.rbegin();
}
int rtop()
{
    assert(!lar.empty());
    return *lar.begin();
}
void lpop()
{
    lsum-=ltop();
    lit.erase(prev(lit.end()));
}
void rpop()
{
    rsum-=rtop();
    lar.erase(lar.begin());
}
void lera(int x)
{
    lsum-=x;
    lit.erase(lit.lower_bound(x));
}
void rera(int x)
{
    rsum-=x;
    lar.erase(lar.lower_bound(x));
}
void lpush(int x)
{
    lit.emplace(x);
    lsum+=x;
}
void rpush(int x)
{
    lar.emplace(x);
    rsum+=x;
}
void balance()
{
    while(lit.size()<lar.size())
    {
        lpush(rtop());
        rpop();
    }
    while(lit.size()>lar.size()+1)
    {
        rpush(ltop());
        lpop();
    }
}
int med()
{
    return ltop();
}
i64 query()
{
    return rsum-lsum+(lit.size()>lar.size()?med():0);
}

void add(int i)
{
    if(lit.empty())
    {
        lpush(a[i]);
    }
    else
    {
        if(ltop()>a[i]) lpush(a[i]);
        else rpush(a[i]);
    }
    balance();
}

void del(int i)
{
    if(lit.count(a[i]))
    {
        lera(a[i]);
    }
    else
    {
        assert(lar.count(a[i]));
        rera(a[i]);
    }
    balance();
}



main()
{
    int n,Q; cin>>n>>Q;
    read_range(a,a+n);
    Mo mo(n,add,del);
    for(i32 q=0; q<Q; ++q)
    {
        int a,b; cin>>a>>b;
        mo.insert(--a,b);
    }
    mo.build();
    vector<i64> ans(Q);
    for(i32 q=0; q<Q; ++q)
    {
        int idx=mo.process();
        ans[idx]=query();
        // dump(idx,med(),lsum,rsum,query());
    }
    for(i64 x : ans)
    {
        cout << x << "\n";
    }
}

// https://yukicoder.me/problems/no/924