《挑战程序设计竞赛》3.3.1 数据结构-树状数组 POJ3468 1804 1990 3109 2155 2886（4）

---

POJ3468 成段更新，区间求和

http://poj.org/problem?id=3468

题意

给你N个数，Q个操作，操作有两种，‘Q a b ’是询问a~b这段数的和，‘C a b c’是把a~b这段数都加上c。

思路

此题是《挑战》书中例题，线段树和树状数组都可以用，我的测试表明树状数组效率略高一些。

主要思想是维护两个树状数组。根据要求提前推导出对应于结果的映射关系。

另外线段树的实现代码见我的博客的3.3.2节。

代码（树状数组）

Source Code

Problem: 3468       User: liangrx06
Memory: 2192K       Time: 1032MS
Language: C++       Result: Accepted
Source Code
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 100000;

typedef long long LL;

int n;
int a[N];
LL bit[2][N+1];

void add(int b, int i, int x)
{
    while (i <= n) {
        bit[b][i] += x;
        i += i&-i;
    }
}

LL sum(int b, int i)
{
    LL s = 0;
    while (i > 0) {
        s += bit[b][i];
        i -= i&-i;
    }
    return s;
}

LL f(int i)
{
    return sum(0, i) + sum(1, i)*i;
}

int main(void)
{
    int q;
    cin >> n >> q;
    memset(bit, 0, sizeof(bit));
    for (int i = 0; i < n; i ++) {
        scanf("%d", &a[i]);
        add(0, i+1, a[i]);
    }

    char c[2];
    int l, r, x;
    while (q --) {
        scanf("%s", c);
        if (c[0] == 'Q') {
            scanf("%d%d", &l, &r);
            printf("%lld\n", f(r)-f(l-1));
        } else {
            scanf("%d%d%d", &l, &r, &x);
            add(0, l, -x*(l-1));
            add(1, l, x);
            add(0, r+1, x*r);
            add(1, r+1, -x);
        }
    }

    return 0;
}

---

POJ1804 逆序数

http://poj.org/problem?id=1804

题意

求快速排序的交换次数，其实就是求逆序数。

思路

逆序数目前已知最常见的方法是利用归并排序来求。另外还可以用线段树和树状数组来求。都是O(nlog
n)的复杂度。
树状数组的实现需要将原数范围[-1000000, 1000000]映射到[1, 2000001]上，这样才符合树状数组的定义，并且不影响逆序数结果。
本人实现了归并排序和树状数组两种方法。这里分别给出代码。

代码1（树状数组）

Source Code

Problem: 1804       User: liangrx06
Memory: 8064K       Time: 422MS
Language: C++       Result: Accepted
Source Code
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 1000;
const int ADD = 1000001;
const int M = ADD*2-1;

int n;
int a[N];

int bit[M+1];

int sum(int i)
{
    int s = 0;
    while (i > 0) {
        s += bit[i];
        i -= i&-i;
    }
    return s;
}

void add(int i, int x)
{
    while (i <= M) {
        bit[i] += x;
        i += i&-i;
    }
}

int main(void)
{
    int t;
    cin >> t;
    for (int k = 1; k <= t; k ++) {
        cin >> n;
        int ans = 0;
        memset(bit, 0, sizeof(bit));
        for (int i = 0; i < n; i ++) {
            scanf("%d", &a[i]);
            a[i] += ADD;
            ans += i - sum(a[i]);
            add(a[i], 1);
        }
        printf("Scenario #%d:\n%d\n\n", k, ans);
    }

    return 0;
}

代码2（归并排序）

Source Code

Problem: 1804       User: liangrx06
Memory: 280K        Time: 32MS
Language: C++       Result: Accepted
Source Code
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;

const int N = 1000;

int n;
int a[N];

int merge_count(int l, int r)
{
    if (r == l) return 0;

    int res = 0;
    int m = (l + r) / 2;
    res += merge_count(l, m);
    res += merge_count(m+1, r);

    int i = l, j = m+1, k = l;
    int b[N];
    while (i <= m || j <= r) {
        if (i > m)
            b[k++] = a[j++];
        else if (j > r) {
            b[k++] = a[i++];
            res += (j-m-1);
        } else if (a[i] <= a[j]) {
            b[k++] = a[i++];
            res += (j-m-1);
        } else
            b[k++] = a[j++];
    }
    for (k = l; k <= r; k ++)
        a[k] = b[k];
    return res;
}

int main(void)
{
    int t;
    cin >> t;
    for (int k = 1; k <= t; k ++) {
        cin >> n;
        for (int i = 0; i < n; i ++)
            scanf("%d", &a[i]);
        printf("Scenario #%d:\n%d\n\n", k, merge_count(0, n-1));
    }

    return 0;
}

---

#
http://poj.org/problem?id=

题意

思路

代码