hdu5792 World is Exploding(多校第五场）树状数组求逆序对 离散化

题目地址：http://acm.hdu.edu.cn/showproblem.php?pid=5792

题目描述：给你n个值，每个值用A[i]表示，然后问你能否找到多少组（a,b,c,d）四个编号，四个编号互不相同，然后a < b, c < d，a代表的值小于b代表的值，c代表的值大于d代表的值。

解题思路：先考虑a和b这两个编号，遍历每一个编号作为b，然后找到b前面有多少个小于b的值，就是对于这一个编号b合理的编号a，对于每一组a和b，就可以考虑c和d，能够满足条件c和d的很显然就是除去a和b整个序列有多少个逆序对，对吧。然后对于每一个b有x个a满足条件，所以对于当前b，能够满足条件的就是（x*所有的逆序对的个数 - 逆序对中包涵a和b的个数）。所有逆序对很容易就可以用树状数组求出来，x也就是b前面有多少个小于编号b代表的值得个数，也很容易求出来，然后包涵a和b个数的逆序对呢，a前面大于A[a]的个数和a后面小于A[a]的个数，b也是一样的。

设x为对于每个b满足条件的a的个数，na代表和a有关系的逆序对,nb代表和b有关系的逆序对，tot代表所有的逆序对个数，然后对于每一个b就有 tot*x - nb*x - （所有na的值）。

 

代码：

#include<cstdio>#include<cmath>#include<iostream>#include<algorithm>#include<vector>#include<stack>#include<cstring>#include<queue>#include<set>#include<string>#include<map>#define inf 9223372036854775807#define INF 9e7+5#define PI acos(-1)using namespace std;typedef long long ll;typedef double db;const int maxn = 5e4 + 5;const int mod = 1e9 + 7;const db eps = 1e-9;ll n, c[maxn], a[maxn], num[maxn], tot, cc[maxn], lar[maxn], b[maxn];struct pos {    ll a, ord, la;} q[maxn];void init() {    tot = 0;    memset(c, 0, sizeof(c));    memset(cc, 0, sizeof(cc));    memset(lar, 0, sizeof(lar)); //这个数组好像不用初始化}ll lowbit(ll x) {    return x & (-x); //求最低位}void updata(ll x, ll p, ll u[]) { //更新    while (x <= maxn) {        u[x] += p;        x += lowbit(x);    }}ll getsum(ll x, ll u[]) {    ll ans = 0;    while (x > 0) {        ans += u[x];        x -= lowbit(x);    }    return ans;}bool cmp1(const pos&a, const pos&b){    return a.a<b.a;}bool cmp2(const pos&a, const pos&b){    return a.ord<b.ord;}void solve() {    while (cin >> n) {        init();        for (int i = 1; i <= n; i++) {            scanf("%I64d", &q[i].a);            q[i].ord = i;        }        int pp = 0; sort(q+1, q+1+n, cmp1);        q[0].a = -1;        for (int i = 1; i <= n; i++) { //离散化            if (q[i].a != q[i-1].a) q[i].la = ++pp;            else {                q[i].la = pp;            }        }        sort(q+1, q+1+n, cmp2);        for (int i = 1; i <= n; i++) {            lar[i] = i - getsum(q[i].la, c) - 1; //求出每个数前面            updata(q[i].la, 1, c);              //有多少比他大的数        }        memset(c, 0, sizeof(c));        for (int i = n; i >= 1; i--) {            num[i] = getsum(q[i].la-1, c);          //求出每个数的            updata(q[i].la, 1, c); tot += num[i]; //后面比他小的个数        }        ll ans = 0; memset(c, 0, sizeof(c));        for (int i = 1; i <= n; i++) {            ll tmp = getsum(q[i].la-1, c);   //对当前b满足条件的a的个数            updata(q[i].la, 1, c);              ll tt = getsum(q[i].la-1, cc);  //求出和a有关系的逆序对            updata(q[i].la, num[i] + lar[i], cc);//num[i]+lar[i]就是和a有关系的逆序对            ans += tmp * tot - tt - (num[i]+lar[i])*tmp;//这里的num[i]+lar[i]是和b有关系的逆序对        }        cout << ans << endl;    }}int main(){    //cin.sync_with_stdio(false);    //freopen("tt.txt", "r", stdin);    //freopen("isharp.out", "w", stdout);    solve();    return 0;}