ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

P1908 逆序对 ——树状数组求逆序对模板

2021-10-02 13:58:07  阅读:137  来源: 互联网

标签:const 树状 int P1908 long 逆序 include define


​​​​​​P1908

求逆序对,注意可能有重复数字,排序的时候要注意,否则会wa

C[x]表示1-x中已经有的数字数量

数组C中存的是正序对,所以每次操作要sum += i - sum(node[i].index),这样才是逆序对

时间复杂度O(nlogn)

// Decline is inevitable,
// Romance will last forever.
// P1908
//#include <bits/stdc++.h>
#include <iostream>
#include <cmath>
#include <cstring>
#include <string>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <deque>
#include <vector>
using namespace std;
#define mst(a, x) memset(a, x, sizeof(a))
#define INF 0x3f3f3f3f
//#define mp make_pair
#define pii pair<int,int>
#define fi first
#define se second
#define ll long long
//#define int long long
const int maxn = 5e5 + 10;
const int maxm = 25;
const int P = 1e9 + 7;
int C[maxn], n, m;
struct Node {
    int v, index;
    bool operator <(const Node &x) const {
        if(v == x.v) return index < x.index;    //有重复数字则不能省略这句话
        return v < x.v;
    }
}node[maxn];
int lb(int x) {
    return x & -x;
}
int sum(int x) {    //前x组的和
    int ret = 0;
    while(x) {
        ret += C[x];
        x -= lb(x);
    }
    return ret;
}
void add(int x, int d) {    //第x组加d
    while(x <= n) {
        C[x] += d;
        x += lb(x);
    }
}
int t = 0;
void solve() {
    scanf("%d", &n);
    for(int i = 1; i <= n; i++) {
        int a;
        scanf("%d", &a);
        node[i].index = i;
        node[i].v = a;
    }
    sort(node + 1, node + n + 1);
    ll ans = 0;
    for(int i = 1; i <= n; i++) {
        add(node[i].index, 1);
        ans += i - sum(node[i].index);
    }
    printf("%lld\n", ans);
}
signed main() {
//    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
//    int T; scanf("%d", &T); while(T--)
//    int T; cin >> T; while(T--)
    solve();
    return 0;
}

标签:const,树状,int,P1908,long,逆序,include,define
来源: https://blog.csdn.net/m0_59273843/article/details/120585356

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有