赛前再巩固一下线段树,又做了一遍kuangbin线段树专题
专题链接:kuangbin专题题目一览
1. HDU 1166 敌兵布阵
区间维护和,单点修改
#include <stdio.h>
using namespace std;
const int MAXN=1e5 + 10;
int origin[MAXN], tree[MAXN<<2];
void pushup(int p) {
tree[p] = tree[p << 1] + tree[p << 1 | 1];
}
void build(int p, int l, int r) {
if (l == r) {
tree[p] = origin[l];
return;
}
int mid = (l + r) >> 1;
build(p << 1, l, mid);
build(p << 1 | 1, mid + 1, r);
pushup(p);
}
void update_node(int p, int l,int r, int q,int v) {
if (l == r) { //查询到点
tree[p] += v;
return;
}
int mid = (l + r) >> 1;
if (q > mid) update_node(p << 1 | 1, mid + 1, r, q, v);
else update_node(p << 1, l, mid, q, v);
pushup(p);
}
int query(int p, int l, int r, int ql, int qr) {
if (ql <= l && r <= qr) return tree[p]; //被包含在询问区域内的区间(有效的部分)
int mid = (l + r) >> 1;
int temp = 0;
if (qr > mid) temp += query(p << 1 | 1, mid + 1, r, ql, qr); //分块切割出有效的部分(已忽略无效部分)
if (ql <= mid) temp += query(p << 1, l, mid, ql, qr);
return temp;
}
int main () {
int T;
scanf("%d", &T);
for (int k = 1; k <= T; k++) {
int n, i, j;
char s[10];
scanf("%d", &n);
for (int i = 1; i <= n; i++)
scanf("%d", origin + i);
build(1, 1, n);
printf("Case %d:\n", k);
while(scanf("%s", s) && s[0] != 'E') {
scanf("%d%d", &i, &j);
if (s[0] == 'A')
update_node(1, 1, n, i, j);
else if (s[0] == 'S')
update_node(1, 1, n, i, -j);
else
printf("%d\n", query(1, 1, n, i, j));
}
}
}
2. HDU 1754 I Hate It
维护区间最大值,单点修改
#include <stdio.h>
#include <algorithm>
using namespace std;
const int MAXN = 2e5 + 10;
int origin[MAXN], tree[MAXN<<2], lazy[MAXN<<2];
void pushup(int p) {
tree[p] = max(tree[p << 1], tree[p << 1 | 1]);
}
void build(int p, int l, int r) {
if (l == r) {
tree[p] = origin[l];
return;
}
int mid = (l + r) >> 1;
build(p << 1, l, mid);
build(p << 1 | 1, mid + 1, r);
pushup(p);
}
void update_node(int p, int l,int r, int q,int v) {
if (l == r) { //查询到点
tree[p] = v;
return;
}
int mid = (l + r) >> 1;
if (q > mid) update_node(p << 1 | 1, mid + 1, r, q, v);
else update_node(p << 1, l, mid, q, v);
pushup(p);
}
int query(int p, int l, int r, int ql, int qr) {
if (ql <= l && r <= qr) return tree[p]; //被包含在询问区域内的区间(有效的部分)
int mid = (l + r) >> 1;
int temp = -1;
if (qr > mid) temp = max(temp, query(p << 1 | 1, mid + 1, r, ql, qr)); //分块切割出有效的部分(已忽略无效部分)
if (ql <= mid) temp = max(temp, query(p << 1, l, mid, ql, qr));
return temp;
}
int main() {
int n, m;
while(scanf("%d%d", &n, &m) != EOF) {
for (int i = 1; i <= n; i++)
scanf("%d", origin + i);
build(1, 1, n);
while(m--) {
char c[3];
int i, j;
scanf("%s%d%d", c, &i, &j);
if (c[0] == 'Q')
printf("%d\n", query(1, 1, n, i, j));
else
update_node(1, 1, n, i, j);
}
}
}
3. POJ 3468 A Simple Problem with Integers
维护区间和,区间修改
#include <stdio.h>
using namespace std;
typedef long long ll;
const int MAXN=1e5 + 10;
ll origin[MAXN], tree[MAXN<<2], lazy[MAXN<<2];
void pushup(int p) {
tree[p] = tree[p << 1] + tree[p << 1 | 1];
}
void pushdown(int p, int l, int r) {
if (lazy[p]) {
lazy[p << 1] += lazy[p];
lazy[p << 1 | 1] += lazy[p];
int mid = (l + r) >> 1;
tree[p << 1] += lazy[p] * (mid - l + 1); //左右儿子结点均按照需要加的值总和更新结点信息
tree[p << 1 | 1] += lazy[p] * (r - mid);
lazy[p] = 0;
}
}
void build(int p, int l, int r) {
if (l == r) {
tree[p] = origin[l];
return;
}
int mid = (l + r) >> 1;
build(p << 1, l, mid);
build(p << 1 | 1, mid + 1, r);
pushup(p);
}
void update(int p, int l, int r, int ql, int qr, ll v) { //区间更新
if (ql <= l && r <= qr) { //当前区间在更新区间内
tree[p] += v * (r - l + 1);
lazy[p] += v;
// pushdown(p, l, r);
return;
}
int mid = (l + r) >> 1;
pushdown(p, l, r);
if (ql <= mid) update(p << 1, l, mid, ql, qr, v);
if (qr > mid) update(p << 1 | 1, mid + 1, r, ql, qr, v);
pushup(p);
}
ll query(int p, int l, int r, int ql, int qr) {
if (ql <= l && r <= qr) return tree[p]; //被包含在询问区域内的区间(有效的部分)
int mid = (l + r) >> 1;
pushdown(p, l, r); //有区间更新时才需要
ll temp = 0;
if (qr > mid) temp += query(p << 1 | 1, mid + 1, r, ql, qr); //分块切割出有效的部分(已忽略无效部分)
if (ql <= mid) temp += query(p << 1, l, mid, ql, qr);
return temp;
}
int main () {
int n, m;
while(scanf("%d%d", &n, &m)!= EOF) {
int i, j;
ll c;
char s[5];
for (int i = 1; i <= n; i++)
scanf("%lld", origin + i);
build(1, 1, n);
while(m--) {
scanf("%s", s);
if (s[0] == 'Q') {
scanf("%d%d", &i, &j);
printf("%lld\n", query(1, 1, n, i, j));
}
else {
scanf("%d%d%lld", &i, &j, &c);
update(1, 1, n, i, j, c);
}
}
}
}
4. POJ 2528 Mayor's posters
见题解:POJ 2528 Mayor's posters [ 离散化 + 线段树 ]
5. HDU 1698 Just a Hook
维护区间和,区间修改
#include <stdio.h>
#include <memory.h>
using namespace std;
const int MAXN=1e5 + 10;
int origin[MAXN], tree[MAXN<<2], lazy[MAXN<<2];
void pushup(int p) {
tree[p] = tree[p << 1] + tree[p << 1 | 1];
}
void pushdown(int p, int l, int r) {
if (lazy[p]) {
lazy[p << 1] = lazy[p];
lazy[p << 1 | 1] = lazy[p];
int mid = (l + r) >> 1;
tree[p << 1] = lazy[p] * (mid - l + 1); //左右儿子结点均按照需要加的值总和更新结点信息
tree[p << 1 | 1] = lazy[p] * (r - mid);
lazy[p] = 0;
}
}
void build(int p, int l, int r) {
if (l == r) {
tree[p] = origin[l];
return;
}
int mid = (l + r) >> 1;
build(p << 1, l, mid);
build(p << 1 | 1, mid + 1, r);
pushup(p);
}
void update(int p, int l, int r, int ql, int qr, int v) { //区间更新
if (ql <= l && r <= qr) { //当前区间在更新区间内
tree[p] = v * (r - l + 1);
lazy[p] = v;
return;
}
int mid = (l + r) >> 1;
pushdown(p, l, r);
if (ql <= mid) update(p << 1, l, mid, ql, qr, v);
if (qr > mid) update(p << 1 | 1, mid + 1, r, ql, qr, v);
pushup(p);
}
int query(int p, int l, int r, int ql, int qr) {
if (ql <= l && r <= qr) return tree[p]; //被包含在询问区域内的区间(有效的部分)
int mid = (l + r) >> 1;
pushdown(p, l, r); //有区间更新时才需要
int temp = 0;
if (qr > mid) temp += query(p << 1 | 1, mid + 1, r, ql, qr); //分块切割出有效的部分(已忽略无效部分)
if (ql <= mid) temp += query(p << 1, l, mid, ql, qr);
return temp;
}
int main () {
int T;
scanf("%d", &T);
for (int k = 1; k <= T; k++) {
int n, m;
scanf("%d%d", &n, &m);
int i, j, c;
for (int i = 1; i <= n; i++)
origin[i] = 1;
memset(lazy, 0, sizeof(lazy));
build(1, 1, n);
while(m--) {
scanf("%d%d%d", &i, &j, &c);
update(1, 1, n, i, j, c);
}
printf("Case %d: The total value of the hook is %d.\n", k, query(1,1,n,1, n));
}
}
6. ZOJ 1610 Count the Colors
见题解:ZOJ 1610 Count the Colors [ 线段树 + 区间染色 ]
7. POJ 3264 Balanced Lineup
维护区间最小值、最大值,不需要更新。用RMQ做更快
#include <stdio.h>
#include <algorithm>
using namespace std;
const int MAXN = 5e4 + 10;
const int INF = 0x3f3f3f3f;
int origin[MAXN];
struct node {
int max;
int min;
}tree[MAXN<<2];
void pushup(int p) {
tree[p].max = max(tree[p << 1].max, tree[p << 1 | 1].max);
tree[p].min = min(tree[p << 1].min, tree[p << 1 | 1].min);
}
void build(int p, int l, int r) {
if (l == r) {
tree[p].max = tree[p].min = origin[l];
return;
}
int mid = (l + r) >> 1;
build(p << 1, l, mid);
build(p << 1 | 1, mid + 1, r);
pushup(p);
}
node query(int p, int l, int r, int ql, int qr) {
if (ql <= l && r <= qr) return tree[p]; //被包含在询问区域内的区间(有效的部分)
int mid = (l + r) >> 1;
node temp, temp2;
temp.max = -INF;
temp.min = INF;
if (qr > mid) {
//一定要注意:这里应该先存下temp2,而不是每次都query出来,可以省很多时间,否则会TLE
temp2 = query(p << 1 | 1, mid + 1, r, ql, qr);
temp.max = max(temp.max, temp2.max);
temp.min = min(temp.min, temp2.min);
}
if (ql <= mid) {
temp2 = query(p << 1, l, mid, ql, qr);
temp.max = max(temp.max, temp2.max);
temp.min = min(temp.min, temp2.min);
}
return temp;
}
int main () {
int n, q;
scanf("%d%d", &n, &q);
for (int i = 1; i <= n; i++) //origin数组的下标从1开始
scanf("%d", origin + i);
build(1, 1, n); //拿区间 1, n 的数据建树
while (q--) {
int i, j;
scanf("%d%d", &i, &j);
node temp = query(1, 1, n, i, j);
printf("%d\n", temp.max - temp.min);
}
}