算法思路
题目:给出策略在一系列可能动态变化(可增可减)的数据中获取最小值,并对你的策略或方案进行分析与评价。
- 动态变化的过程则考虑可能从一串数据的中间删除某个数——如此看来栈和队列难以满足要求
- 由于近日竞赛算法课学习到的线段树数据结构,可以求得一串数组任意区间中的最小值,并且完成更新等操作——考虑将那一份代码修改后用于本题题解
表示思路
-
准备三个数组,a[N]用于存所有动态的数据,置于叶子节点处;minv[4N]存放每个对应区间的最小值;addv[4N]用于懒标记,更新某个节点
-
动态数据-增添数据则直接增加节点,删除数据则把对应叶子节点数值改成无穷大;本数据结构可以返回任意区间的最小值,在本题中每次都直接返回根节点数值即可
-
构建的线段树如下
-
包含模块函数
- int min(int a, int b) ;
- 封装一个求最小值的方法
- void pushup(int x) ;
- 递归的方法从叶子节点到根节点完成minv[x]的构建
- void pushdown(int x);
- 递归的方法从根节点到叶子结点完成addv[x]的构建
- void build(int x, int left, int &right) ;
- 建树方法
- void change(int x, int left, int &right, int q, int v);
- 更新的方法
- void add(int y);
- 增加数据的简要封装
- int min(int a, int b) ;
完整代码
//算法第三题 动态求最小值 每次直接返回根节点数值就可以了
//需要加一个添加数字的方法
#include<string.h>
#include<stdio.h>
#define N 1000
#define inf 1000000
int a[N];
int minv[N * 4];
int addv[N * 4]; //这个节点也已经完成了更新,但是加上标记的意思是下面的节点都需要加这个数
int min(int a, int b) {
return (a > b ? b : a);
}
void pushup(int x) { //把下标x节点的val设置成对应区间最小值
minv[x] = min(minv[2 * x], minv[2 * x + 1]);
}
void pushdown(int x) { //对这个x进行检查+更新
if (addv[x] == 0) return;
addv[x * 2] = addv[x * 2] + addv[x];
addv[2 * x + 1] += addv[x];
minv[x * 2] += addv[x];
minv[x * 2 + 1] += addv[x]; //对左右子节点的minv
addv[x] = 0;
}
//x是根节点的下标,left是左下标,right是右下标
void build(int x, int left, int &right) {
if (left == right) {
minv[x] = a[left];
return;
}
int mid = (left + right) >> 1;
build(x * 2, left, mid);
build(x * 2 + 1, mid + 1, right);
pushup(x);
}
//更新,只要在build函数上的基础上修改就行,也是从底层往上搭建
//更新某一个值
void change(int x, int left, int &right, int q, int v) { //left right 其实也就是用来查找了 不断缩小 类似二分
if (left == right) {
minv[x] += v;
addv[x] += v;
return;
}
int mid = (left + right) >> 1;
pushdown(x);//pushdown就是对于这个点完成更新,同时对左右孩子做好标记
if (q <= mid) {
change(x * 2, left, mid, q, v);
} else {
change(x * 2 + 1, mid + 1, right, q, v);
}
pushup(x);
}
void add(int y) {
int i;
for (i = 1; a[i] != 0; i++); //i到了可以添加的地方
a[i] = y;
}
int main() {
memset(addv, 0, N * 4 - 1);
printf("初始化数据有几个?\n");
int c;
scanf("%d", &c);
printf("请依次输入数值\n");
for (int i = 1; i <= c; i++) {
scanf("%d", &a[i]);
}
build(1, 1, c);
printf("初始化成功!\n");
printf("测试!请选择你要进行的操作\n1.查询最小值\n2.更新\n3.添加数据\n4.删除数据\n");
int op;
while (scanf("%d", &op) != EOF) {
switch (op) {
case 1: {
printf("最小值是%d\n", minv[1]);
break;
}
case 2: {
printf("把下标为e的节点更新为f,请输入e,f\n");
int e, f;
scanf("%d %d", &e, &f);
int value = f - a[e];
a[e] = f;
change(1, 1, c, e, value);
printf("更新成功!\n");
for (int i = 1; i <= c; i++) {
printf("%d ", a[i]);
}
printf("\n");
break;
}
case 3: {
printf("输入增加的数据\n");
int y;
scanf("%d", &y);
add(y);
c++;
build(1, 1, c);
printf("添加成功!\n");
break;
}
case 4: {
printf("输入想要删除的数据\n");
int del;
scanf("%d", &del);
int i;
for (i = 1; a[i] != del; i++);
a[i] = inf;
change(1, 1, c, i, inf);
printf("删除成功!\n");
break;
}
}
}
}
测试结果展示
总结与讨论
-
思路总结
- 线段树用于求区间的最小值尤为方便,父节点记录对应区间的最小值可以起到存储数值的作用