Codeforces-1677A: Tokitsukaze and Strange Inequality 【动态规划、区间数据结构】

Codeforces-1677A: Tokitsukaze and Strange Inequality

题目传送门： Codeforces-1677A

题目

题目截图

样例描述

题目大意

  给定一个长度为 $n$ 的排列 $p_1,p_2,\cdots ,p_n$，目标是找到满足 $p_a<p_c\text{ and }{p}_b>p_d\text{ and }a<b<c<d$ 的四元组 $a,b,c,d$ 的个数。

题目解析

  题目需要在 $O(n^2)$ 的复杂度左右解出，因此我们可以用双层循环。
  首先确定遍历顺序，我们遍历 $c$，之后我们遍历 $a<c$。那么我们需要做的就是统计逆序对 $(b,d),s.t.a<b<c<d$ 的数量。逆序对的求法很简单，因此本问题的核心问题是如何控制逆序对的起点在 $(a,c)$，而逆序对的终点在 $(c,n]$。
  设 $f_i$ 代表整个排列中以 $i$ 为起点的逆序对个数。当 $a<i<c$ 那么我们可以处理出终点 $<c$ 的逆序对数，那么 $f_i$ 将 $[i,c]$ 的逆序对减去，就是起点在 $i$，而逆序对的终点在 $(c,n]$ 的逆序对数了。处理的方法可以随着遍历进行，因为起始点和终止点在遍历过程中都在 $[1,c]$ 的在之后也用不到了，所以把 $f_i$ 随着 $c$ 的增长减去 $a_i>a_c$ 的情况就可以动态更新了，使用二维数组进行预处理也是可以的。有了这个值，我们利用其前缀和就可以对 $a$ 进行遍历了。整个算法是 $O(n^2)$。

Code

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int maxn = 5005;
int A[maxn], f[maxn], p[maxn], pre[maxn];

int lowbit(int x) {return (x & -x);}
void add(int x) {
    while(x < maxn) ++f[x], x += lowbit(x);
}
int query(int x) {
    int ret = 0;
    while(x > 0) {
        ret += f[x];
        x -= lowbit(x);
    }
    return ret;
}

int main() {
    int t, n;
    cin >> t;
    while(t--) {
        cin >> n;
        memset(f, 0, sizeof(f));
        for(int i=1; i<=n; ++i)
            cin >> A[i];
        for(int i=n; i>=1; --i) {
            p[i] = query(A[i]);
            add(A[i]);
        }

        LL ans = 0;
        for(int c=2; c<=n; ++c) {
            for(int b=1; b<c; ++b)
                if(A[b] > A[c]) --p[b];
            pre[0] = 0;
            for(int b=1; b<c; ++b)
                pre[b] = pre[b-1] + p[b];
            for(int a=1; a<c; ++a) {
                if(A[a] < A[c])
                    ans += pre[c-1] - pre[a];
            }
        }
        cout << ans << endl;
    }
    return 0;
}