Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include<bits/stdc++.h>
- using namespace std;
- #ifdef LOCAL
- #include "../debug.h"
- #else
- #define dbg(...)
- #endif
- template<class Info, class Tag>
- struct LazySegmentTree {
- int n;
- std::vector<Info> info;
- std::vector<Tag> tag;
- LazySegmentTree() : n(0) {}
- LazySegmentTree(int n_, Info v_ = Info()) {
- init(n_, v_);
- }
- template<class T>
- LazySegmentTree(std::vector<T> init_) {
- init(init_);
- }
- void init(int n_, Info v_ = Info()) {
- init(std::vector(n_, v_));
- }
- template<class T>
- void init(std::vector<T> init_) {
- n = init_.size();
- info.assign(4 << std::__lg(n), Info());
- tag.assign(4 << std::__lg(n), Tag());
- std::function<void(int, int, int)> build = [&](int p, int l, int r) {
- if (r - l == 1) {
- info[p] = init_[l];
- return;
- }
- int m = (l + r) / 2;
- build(2 * p, l, m);
- build(2 * p + 1, m, r);
- pull(p);
- };
- build(1, 0, n);
- }
- void pull(int p) {
- info[p] = info[2 * p] + info[2 * p + 1];
- }
- void apply(int p, const Tag &v) {
- if(v.add == -1) return;
- info[p].apply(v);
- tag[p].apply(v);
- }
- void push(int p) {
- apply(2 * p, tag[p]);
- apply(2 * p + 1, tag[p]);
- tag[p] = Tag();
- }
- void modify(int p, int l, int r, int x, const Info &v) {
- if (r - l == 1) {
- info[p] = v;
- return;
- }
- int m = (l + r) / 2;
- push(p);
- if (x < m) {
- modify(2 * p, l, m, x, v);
- } else {
- modify(2 * p + 1, m, r, x, v);
- }
- pull(p);
- }
- void modify(int p, const Info &v) {
- modify(1, 0, n, p, v);
- }
- Info rangeQuery(int p, int l, int r, int x, int y) {
- if (l >= y || r <= x) {
- return Info();
- }
- if (l >= x && r <= y) {
- return info[p];
- }
- int m = (l + r) / 2;
- push(p);
- return rangeQuery(2 * p, l, m, x, y) + rangeQuery(2 * p + 1, m, r, x, y);
- }
- Info rangeQuery(int l, int r) {
- return rangeQuery(1, 0, n, l, r);
- }
- void rangeApply(int p, int l, int r, int x, int y, const Tag &v) {
- if (l >= y || r <= x) {
- return;
- }
- if (l >= x && r <= y) {
- apply(p, v);
- return;
- }
- int m = (l + r) / 2;
- push(p);
- rangeApply(2 * p, l, m, x, y, v);
- rangeApply(2 * p + 1, m, r, x, y, v);
- pull(p);
- }
- void rangeApply(int l, int r, const Tag &v) {
- return rangeApply(1, 0, n, l, r, v);
- }
- template<class F>
- int findFirst(int p, int l, int r, int x, int y, F &&pred) {
- if (l >= y || r <= x) {
- return -1;
- }
- if (l >= x && r <= y && !pred(info[p])) {
- return -1;
- }
- if (r - l == 1) {
- return l;
- }
- int m = (l + r) / 2;
- push(p);
- int res = findFirst(2 * p, l, m, x, y, pred);
- if (res == -1) {
- res = findFirst(2 * p + 1, m, r, x, y, pred);
- }
- return res;
- }
- template<class F>
- int findFirst(int l, int r, F &&pred) {
- return findFirst(1, 0, n, l, r, pred);
- }
- template<class F>
- int findLast(int p, int l, int r, int x, int y, F &&pred) {
- if (l >= y || r <= x) {
- return -1;
- }
- if (l >= x && r <= y && !pred(info[p])) {
- return -1;
- }
- if (r - l == 1) {
- return l;
- }
- int m = (l + r) / 2;
- push(p);
- int res = findLast(2 * p + 1, m, r, x, y, pred);
- if (res == -1) {
- res = findLast(2 * p, l, m, x, y, pred);
- }
- return res;
- }
- template<class F>
- int findLast(int l, int r, F &&pred) {
- return findLast(1, 0, n, l, r, pred);
- }
- };
- struct Tag {
- int add = -1;
- void apply(Tag t) {
- add = t.add;
- }
- };
- struct Info {
- int64_t min = 0;
- int sz;
- Info(int mn = 0, int s = 1) {
- min = mn;
- sz = s;
- }
- void apply(Tag t) {
- min = sz * 1ll * t.add;
- }
- };
- Info operator+(const Info &a, const Info &b) {
- return {a.min + b.min, a.sz + b.sz};
- }
- int32_t main() {
- ios_base::sync_with_stdio(0);
- cin.tie(0);
- auto __solve_testcase = [&](int test) {
- int N; cin >> N;
- vector<int> A(N);
- for(auto &a: A)
- cin >> a;
- set<int> gud; gud.insert(-1);
- vector<int> pos(N + 1, -1);
- LazySegmentTree<Info, Tag> segl(N), segr(N);
- int64_t res = 0;
- for(int i = 0 ; i < N ; ++i) {
- segl.rangeApply(i, i + 1, Tag(i + 1));
- segr.rangeApply(i, i + 1, Tag(i + 1));
- int v = A[i];
- if(pos[v] >= 0) {
- int in = *prev(gud.lower_bound(pos[v])); gud.insert(pos[v]);
- dbg(in, pos[v], v);
- segl.rangeApply(in + 1, pos[v] + 1, Tag(pos[v]));
- segr.rangeApply(0, pos[v] + 1, Tag(i + 1));
- }
- pos[v] = i;
- res += segr.rangeQuery(0, i).min - segl.rangeQuery(0, i).min;
- }
- cout << res << '\n';
- };
- int NumTest = 1;
- cin >> NumTest;
- for(int testno = 1; testno <= NumTest ; ++testno) {
- __solve_testcase(testno);
- }
- return 0;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement