数据结构(荣誉)实验五 树状数组

简介: 数据结构(荣誉)实验五 树状数组

1. 树状数组操作


题目描述


给你n个数,创建一个树状数组,并执行相应操作,按格式要求输出操作结果。执行的操作有以下两种形式:


C i dt ,表示更新A[i],使得A[i]=A[i]+dt,其中1<=i<=n;


Q i j ,表示询问区间和,即A[i]+A[i+1]+…+A[j]的值,其中1<=i<=j<=n。


输入


第一行一个正整数n(1<=n<=10000),代表数据个数。


接下来一行是n个数据。


接下来一行一个正整数m,代表m个操作。


接下来m行,每行一个操作,格式如上所述。


输出


第一行输出创建后的树状数组,数据之间以空格分隔。


接下来m行,输出m个操作结果:对每个“C i dt ”操作,输出更新后的树状数组,数据间以空格分隔;对每个“Q i j ”操作,输出一个数值。


样例输入


10

1 3 2 6 7 -2 5 8 4 10

5

Q 1 10

C 2 4

Q 2 6

C 6 -3

Q 6 6


样例输出


1 4 2 12 7 5 5 30 4 14

44

1 8 2 16 7 5 5 34 4 14

20

1 8 2 16 7 2 5 31 4 14

-5


题解

#include<iostream>
using namespace std;
int treeNode[10005] = {0};
int length;
int lowbit(int num) {
    return num & (-num);
}
int query(int index) {
    int ans = 0;
    while (index > 0) {
        ans += treeNode[index];
        index -= lowbit(index);
    }
    return ans;
}
void change(int index, int value) {
    while (index <= length) {
        treeNode[index] += value;
        index += lowbit(index);
    }
}
void display() {
    cout << treeNode[1];
    for (int i = 2; i <= length; i++) {
        cout << " " << treeNode[i];
    }
    cout << endl;
}
int main() {
    cin >> length;
    int *num = new int[length + 1];
    for (int i = 1; i <= length; i++) {
        num[i] = 0;
        treeNode[i] = 0;
    }
    for (int i = 1; i <= length; i++) {
        cin >> num[i];
        change(i, num[i]);
    }
    display();
    int n;
    char ope;
    cin >> n;
    while (n--) {
        cin >> ope;
        if (ope == 'Q') {
            int startIndex, endIndex;
            cin >> startIndex >> endIndex;
            cout << query(endIndex) - query(startIndex - 1) << endl;
        } else {
            int index, value;
            cin >> index >> value;
            change(index, value);
            display();
        }
    }
    return 0;
}

2. 逆序对


题目描述


给你n个数,每个数a[i]都是不超过1 0 9 10^910

9

的非负整数。求其中逆序对的个数,即所有这样的数对(i , j )满足1<=i<j<=n且a[i]>a[j]。要求用树状数组的相关操作完成题目。


输入


第一行一个正数n(1<=n<=100000),表示数据的个数。


接下来一行是n个整数。


输出


第一行一个整数m,代表逆序对的个数。


样例输入


5

4 7 2 10 9


样例输出


3


题解

#include<iostream>
using namespace std;
int treeNode[10005] = {0};
int length;
int lowbit(int num) {
    return num & (-num);
}
int query(int index) {
    int ans = 0;
    while (index > 0) {
        ans += treeNode[index];
        index -= lowbit(index);
    }
    return ans;
}
void change(int index, int value) {
    while (index <= length) {
        treeNode[index] += value;
        index += lowbit(index);
    }
}
void display() {
    cout << treeNode[1];
    for (int i = 2; i <= length; i++) {
        cout << " " << treeNode[i];
    }
    cout << endl;
}
int getValue(int num) {
    return query(num) - query(num - 1);
}
int main() {
    cin >> length;
    int *num = new int[length + 1];
    for (int i = 1; i <= length; i++) {
        num[i] = 0;
        treeNode[i] = 0;
    }
    for (int i = 1; i <= length; i++) {
        cin >> num[i];
        change(i, num[i]);
    }
    int count = 0;
    for (int i = 1; i <= length; i++) {
        for (int j = i + 1; j <= length; j++) {
            if (getValue(i) > getValue(j)) {
                count++;
            }
        }
    }
    cout << count << endl;
    return 0;
}


3.矩阵操作


题目描述


给定一个n✖n的矩阵A,其中每个元素不是0就是1。A[i,j]表示在第i行、第j列的数,初始时,A[i,j]=0 (1<=i,j<=n)。


我们可以按照如下方式改变矩阵:给定一个左上角在(x1,y1)、右下角在(x2,y2)的矩形,通过使用“not”操作改变这个矩形内的所有元素值(元素0变成1,元素1变成0)。为了维护矩阵的信息,你需要写个程序来接收并且执行以下两个操作:


(1)C x1 y1 x2 y2 (1<=x1<=x2<=n, 1<=y1<=y2<=n),表示更新操作,将改变左上角为(x1,y1)、右下角为(x2,y2)的矩形区域内的数据值,若元素值为0,则变成1;若元素值为1,则变成0。


(2)Q x y (1<=x,y<=n),表示询问操作,询问A[x,y]的值。


输入


第一行两个整数n和T(2<=n<=1000, 1<=T<=50000),分别代表方阵大小和操作数。


接下来T行,每行包含一个操作,以“C x1 y1 x2 y2”或者“Q x y”的形式给出,具体描述如上。


输出


对每个询问操作,输出一行一个整数,表示相应矩阵元素的值。


样例输入


2 10

C 2 1 2 2

Q 2 2

C 2 1 2 1

Q 1 1

C 1 1 2 1

C 1 2 1 2

C 1 1 2 2

Q 1 1

C 1 1 2 1

Q 2 1


样例输出


1

0

0

1


提示


一般的,树状数组是处理查询一个区间、修改一个点,而本题是要处理查询一个点、修改一个区间。在修改区间的时候,可以考虑容斥原理。修改时加一个tag,修改偶数次相当于未修改。


题解

#include<iostream>
using namespace std;
int treeNode[1005][1005] = {0};
int length;
int lowbit(int num) {
    return num & (-num);
}
int query(int index) {
    int ans = 0;
    while (index > 0) {
        ans += treeNode[index][index];
        index -= lowbit(index);
    }
    return ans;
}
void change(int index, int value) {
    while (index <= length) {
        treeNode[index][index] += value;
        index += lowbit(index);
    }
}
void display() {
    cout << treeNode[1];
    for (int i = 2; i <= length; i++) {
        cout << " " << treeNode[i];
    }
    cout << endl;
}
int main() {
    cin >> length;
    int n;
    cin >> n;
    for (int i = 1; i <= length; i++) {
        for (int j = i; j <= length; j++) {
            treeNode[i][j] = 0;
        }
    }
    while (n--) {
        string ope;
        cin >> ope;
        if (ope == "C") {
            int x1, y1, x2, y2;
            cin >> x1 >> y1 >> x2 >> y2;
            for (int i = x1; i <= x2; i++) {
                for (int j = y1; j <= y2; j++) {
                    if (treeNode[i][j] == 0) {
                        treeNode[i][j] = 1;
                    } else {
                        treeNode[i][j] = 0;
                    }
                }
            }
        }
        if (ope == "Q") {
            int x, y;
            cin >> x >> y;
            cout << treeNode[x][y] << endl;
        }
    }
    return 0;
}
相关文章
|
13天前
|
存储 机器学习/深度学习 算法
上机实验三 图的最小生成树算法设计 西安石油大学数据结构
上机实验三 图的最小生成树算法设计 西安石油大学数据结构
19 1
|
13天前
|
存储 算法 C语言
上机实验四 图的最小生成树算法设计 西安石油大学数据结构
上机实验四 图的最小生成树算法设计 西安石油大学数据结构
22 1
|
1月前
|
存储 算法 Serverless
【软件设计师备考 专题 】数据结构深度解析:从数组到图
【软件设计师备考 专题 】数据结构深度解析:从数组到图
56 0
|
4月前
|
数据库 索引
认真学习数据结构之树
认真学习数据结构之树
35 0
|
10月前
|
存储 程序员
数据结构学习分享之树的介绍
前面我们学的都是链式结构或数组这种线性结构,今天我们正式开始学习"树"这个结构.树涉及的问题有很多,包括普通树,二叉树,二叉树又分完全二叉树和非完全二叉树,而我们要掌握的结构"堆"其本质就是一种完全二叉树, 所以在开始讲堆之前,我们应该先了解一些树相关的知识
|
存储 移动开发 算法
攻克数据结构和算法——第六天:排序
若关键字是主关键字(关键字值不重复),这无论采用何种排序方法,排出的结果都是唯一的;若关键字是次关键字(关键字值可以重复),则排出的结果可能不唯一。
157 0
攻克数据结构和算法——第六天:排序
|
存储 机器学习/深度学习 算法
数据结构 | 有关树和二叉树的详解【内附考点精析】
树和二叉树的基本概念和性质,内附精致讲解图和推理过程
162 0
数据结构 | 有关树和二叉树的详解【内附考点精析】
|
算法
计算机考研数据结构算法模板
计算机考研数据结构算法模板
122 0
计算机考研数据结构算法模板
|
算法
数据结构上机实践第三周项目1
数据结构上机实践第三周项目1
数据结构上机实践第三周项目1
|
存储 算法
数据结构上机实践第三周项目4(2)—顺序表应用
数据结构上机实践第三周项目4(2)—顺序表应用
数据结构上机实践第三周项目4(2)—顺序表应用