【LG1393】动态逆序对

题面

题解

\(CDQ\)分治，按照时间来分治

应为一个删除不能对前面的操作贡献，所以考虑一个删除操作对它后面时间的操作的贡献

用上一个答案减去次贡献即可

代码

#include <iostream>

#include <cstdio>

#include <cstdlib>

#include <cstring>

#include <cmath>

#include <algorithm>

#include <vector>

using namespace std;

namespace IO {

    const int BUFSIZE = 1 << 20;

    char ibuf[BUFSIZE], *is = ibuf, *it = ibuf;

    inline char gc() {

        if (is == it) it = (is = ibuf) + fread(ibuf, 1, BUFSIZE, stdin);

		return *is++;

    }

}

inline int gi() {

    register int data = 0, w = 1;

    register char ch = 0;

    while (ch != '-' && (ch > '9' || ch < '0')) ch = IO::gc();

    if (ch == '-') w = -1 , ch = IO::gc();

    while (ch >= '0' && ch <= '9') data = data * 10 + (ch ^ 48), ch = IO::gc();

    return w * data;

}

const int MAX_N = 40005;

struct Node { int x, y, z, s; } t[MAX_N];

bool cmp_x(Node a, Node b) { return a.x < b.x; }

bool cmp_y(Node a, Node b) { return a.y < b.y; }

int N, M, ans, X[MAX_N], a[MAX_N], b[MAX_N], c[MAX_N], d[MAX_N];

inline int lb(int x) { return x & -x; }

void add(int x, int v) { while (x <= N) c[x] += v, x += lb(x); }

int sum(int x) { int res = 0; while (x > 0) res += c[x], x -= lb(x); return res; }

void Div(int l, int r) {

	if (l == r) return ;

	int mid = (l + r) >> 1;

	Div(l, mid); Div(mid + 1, r);

	int j = mid + 1;

	for (int i = l; i <= mid; i++) {

	    for ( ; j <= r && t[j].y < t[i].y; ) add(t[j].z, 1), ++j;

	    t[i].s += sum(N) - sum(t[i].z);

    }

    for (int i = mid + 1; i < j; i++) add(t[i].z, -1);

    j = r;

    for (int i = mid; i >= l; i--) {

        for ( ; j > mid && t[j].y > t[i].y; ) add(t[j].z, 1), --j;

        t[i].s += sum(t[i].z - 1);

    }

    for (int i = r; i > j; i--) add(t[i].z, -1);

    inplace_merge(&t[l], &t[mid + 1], &t[r + 1], cmp_y);

}

int main () {

    N = gi(), M = gi();

    for (int i = 1; i <= N; i++) X[i] = a[i] = gi();

    sort(&X[1], &X[N + 1]);

    for (int i = 1; i <= N; i++) a[i] = lower_bound(&X[1], &X[N + 1], a[i]) - X;

    for (int i = N; i >= 1; i--) ans += sum(a[i] - 1), add(a[i], 1);

    for (int i = 1; i <= N; i++) t[i].x = N + 1, t[i].y = i, t[i].z = a[i];

    for (int i = 1; i <= M; i++) {

        d[i] = gi();

        t[d[i]].x = i;

    }

    sort(&t[1], &t[N + 1], cmp_x);

	memset(c, 0, sizeof(c));

	Div(1, N);

	for (int i = 1; i <= N; i++) c[t[i].y] = t[i].s;

	printf("%d ", ans);

	for (int i = 1; i <= M; i++) printf("%d ", ans -= c[d[i]]);

    return 0;

}

巴特西

【LG1393】动态逆序对

【LG1393】动态逆序对

题面

题解

最新文章

热门文章