18.12.20 DSA 重要逆序对

描述

给定N个数的序列a₁,a₂,...a_N，定义一个数对(a_i, a_j)为“重要逆序对”的充要条件为 i < j 且 a_i > 2a_j。求给定序列中“重要逆序对”的个数。

输入

第一行为序列中数字的个数N（1 ≤ N ≤ 200000）。
第二行为序列a1, a2 ... aN(0 ≤a ≤ 10000000)，由空格分开。输出输出一个整数，为给序列中“重要逆序对”的个数。

样例输入

10
0 9 8 7 6 5 4 3 2 1

样例输出

16

提示

如果使用printf输出long long类型，请用%lld
数据范围
对于40%的数据，有N ≤ 1000。

#include <iostream>
#include <string.h>
#include <algorithm>
#include <stack>
#include <string>
#include <math.h>
#include <queue>
#include <stdio.h>
#include <string.h>
#include <set>
#include <vector>
#include <fstream>
#define maxn 200005
#define inf 999999
#define cha 127
#define eps 1e-6
using namespace std;

long long ans;
long num[maxn],tmp[maxn];
int n;

void mysort(int left,int right) {
    if (left == right)return;
    int mid = (left + right) / 2;
    mysort(left, mid), mysort(mid + 1, right);
    int i = left, j = mid + 1, p = left, t = left;
    while (i <= mid && j <= right) {
        if (num[i] <= num[j])
            tmp[p++]=num[i++];
        else {
            while (num[t] <= 2 * num[j]&&t<=mid)
                t++;
            if (num[t] > 2 * num[j])
                ans += mid - t + 1;
            tmp[p++] = num[j++];
        }
    }
    while (i <= mid)
        tmp[p++] = num[i++];
    while (j <= right)
        tmp[p++] = num[j++];
    for (int i = left; i <= right; i++)
        num[i] = tmp[i];
}

void init() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++)
        scanf("%ld", &num[i]);
    mysort(1, n);
    printf("%lld
", ans);
}

int main() {
    init();
    return 0;
}

好像是把maxn写错了一直TLE……哭了

用同一个指针变量指示有点难想