数据结构
一、线段树
ACwing
1.1275最大数(AcWing)
题目链接:https://www.acwing.com/problem/content/1277/
题意:给定一个有n个数的正整数序列,你可以进行两种操作:1.添加操作:想序列后添加一个数,序列长度变为n+1。2.询问操作:询问这个序列中最后L个数中最大的数是多少。一共有m次操作,输出每次的询问操作的答案。
思路:这道题就只有一个坑点,就是在添加的时候,每次都是(t + a) % p,所以你要用一个变量来存储上一次询问的值即可。其它的就是线段树的模板。接下来说一下这道题涉及到的线段树的几个模板:
1.先说一下线段树的每一个节点所需要存储那些信息,首先是必然包含左儿子和右儿子,然后其他的信息就是根据题意来增加了,在这个题中,它的询问操作是一个范围内的最大值,所以节点还需要存储一个最大值,简单来说,某个节点所存储的信息能够从它的左右儿子推导而来。
struct node
{
int l, r;
int v;
};
2.更新模板:更新某个节点的值。在这个题就是更新某个节点的最大值。
void pushup(int u)
{
tr[u].v = max(tr[u << 1].v, tr[u << 1 | 1].v);
}
3.建树模板:建树的过程就是一个递归的过程,只要是没有到达叶子节点,就一直递归。
void build(int u, int l, int r)
{
tr[u] = {
l, r};
if (l == r) return ;///叶子节点
int mid = l + r >> 1;
build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
}
4.查询模板:这里先贴一个图就比较好理解:
假如我们现在建了一个范围是在1-10的树,那么我们现在要找1 - 6之间的最大值,那么我们是不是要将1-6分为1-5和6这两段里面去找,后面这个我们一眼就能看出是6,但是在代码中我们又怎么实现呢?首先我们要一直记住我们要找的范围是1-6,假如当前的节点的范围被1-6所包含,那么此时我们是不是就该直接返回当前节点的最大值,假如我们现在要求6-10之间的最大值,而我们现在的节点是6-8这个节点,那么此时的最大值是不是6-8的最值,然后找到9-10的最值,两者作比较就得到了最值,那么我们就证明到了包含的这种情况。然后,我们再找当前节点和要寻找的范围之间有没有交集,那边有交集就往那边找。下面我们直接看代码。
int query(int u, int l, int r)
{
if (tr[u].l >= l && tr[u].r <= r) return tr[u].v;
int v = 0;
int mid = tr[u].l + tr[u].r >> 1;
if (l <= mid) v = query(u << 1, l, r);
if (r > mid) v = max(v, query(u << 1 | 1, l, r));
return v;
}
5.修改模板:我们是将x这个节点的值修改为v,就是要注意的一点是,在每次修改完左儿子和右儿子之后不要忘了更新父节点
int query(int u, int l, int r)
{
if (tr[u].l >= l && tr[u].r <= r) return tr[u].v;
int v = 0;
int mid = tr[u].l + tr[u].r >> 1;
if (l <= mid) v = query(u << 1, l, r);
if (r > mid) v = max(v, query(u << 1 | 1, l, r));
return v;
}
AC代码
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N = 200010;
struct node
{
int l, r;
int v;
}tr[N * 4];
int m, p;
void pushup(int u)
{
tr[u].v = max(tr[u << 1].v, tr[u << 1 | 1].v);
}
void build(int u, int l, int r)
{
tr[u] = {
l, r};
if (l == r) return ;
int mid = l + r >> 1;
build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
}
int query(int u, int l, int r)
{
if (tr[u].l >= l && tr[u].r <= r) return tr[u].v;
int v = 0;
int mid = tr[u].l + tr[u].r >> 1;
if (l <= mid) v = query(u << 1, l, r);
if (r > mid) v = max(v, query(u << 1 | 1, l, r));
return v;
}
void modify(int u, int x, int v)
{
if (tr[u].l == x && tr[u].r == x) tr[u].v = v;
else
{
int mid = tr[u].l + tr[u].r >> 1;
if (x <= mid) modify(u << 1, x, v);
else modify(u << 1 | 1, x, v);
pushup(u);
}
}
signed main()
{
int n = 0, last = 0;
scanf("%d%d", &m, &p);
build(1, 1, m);
int x;
char op[2];
while (m --)
{
scanf("%s%d", op, &x);
if (*op == 'Q') {
last = query(1, n - x + 1, n); printf("%d\n", last); }
else {
modify(1, n + 1, (last + x) % p); n ++;}
}
return 0;
}
2.245. 你能回答这些问题吗
题目链接:https://www.acwing.com/problem/content/246/
题意:给你一个长度为n的数列,然后又m条指令,指令有两种:1.1 x y,查询区间[x,y]中的最大连续子段和。2.2 x y,把A[x]改成y。对于每一条查询指令输出查询的答案。
思路:首先考虑每个节点应该包含那些信息,左右儿子l,r肯定是包含的,然后最大连续子段和tmax肯定是包含的,接下来我们想一下最大连续子段和该怎么计算,我们先来看一个图
就上面的三个量,是不够推到的,假设要求的范围是上面的红括号内,那么这个时候的最大连续子段的合适1到左括号+2到右括号,所以还需要添加左范围的以1为结点的后缀和,加上以2为开始的到右括号的前缀和,所以还需要添加的是以lmax(最大后缀和)rmax(最大前缀和),那么求某个点开始的前缀和也会有两种情况,那么是只在前半段内或者包含了右边范围的一点,要么是rmax或者是sum+lmax,所以还需要添加三个变量,sum,lmax,rmax。
struct node
{
int l, r;
int sum, lmax, rmax, tmax;
}tr[N * 4];
接下来我们看更新操作,对于每一个结点我们需要维护的值有四个,sum,tmax,rmax,lmax。现在来看一下lmax怎么计算(rmax也是同理)
假设要求的前缀和是上面1,那么就是lmax,如果是第二种情况,那么就是l_sum + r_lmax,rmax也同理。我们又来看一下tmax怎么更新,现在左右儿子中的找到最大的tmax,或者是l.rmax+r.lmax。
补充:这里的最大前缀和和最大后缀和我们都是以某个节点的终点来找的。
上面我们找到了每个节点应该包含的信息,现在我们来看更新模板,我们要更新的是每个节点的最大连续子段和,那么就是找父节点和它的两个儿子的最大连续子段和的最大值,所以这里应该这样写:
void pushup(node &u, node &l, node &r)
{
u.sum = l.sum + r.sum;
u.lmax = max(l.max, l.max + r.lmax);
u.rmax = max(r.max, r.sum + l.rmax);
u.tmax = max(max(l.tmax, r.tmax), ;.rmax + r.lmax);
}
void push(int u)
{
pushup(tr[u], tr[u << 1], tr[u << 1 | 1]);
}
建树和修改和之前的都差不多,唯一一点区别就是需要更新,我们现在来看一下查找操作,我们在查找的时候要记录四个量,所以我们干脆直接返回一个结构体,我们要考虑三种情况,只在左边,只在右边,或者横穿,前面两种情况好些,但是横穿这种情况,我们要找到左边的,在找到右边的,最后在更新,在返回。看代码:
Node query(int u, int l, int r)
{
if (tr[u].l >= l && tr[u].r <= r) return tr[u];
else
{
int mid = tr[u].l + tr[u].r >> 1;
if (r <= mid) return query(u << 1, l, r);
else if (l > mid) return query(u << 1 | 1, l, r);
else
{
auto left = query(u << 1, l, r);
auto right = query(u << 1 | 1, l, r);
Node res;
pushup(res, left, right);
return res;
}
}
}
AC代码:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 500010;
int n, m;
int w[N];
struct Node
{
int l, r;
int sum, lmax, rmax, tmax;
}tr[N * 4];
void pushup(Node &u, Node &l, Node &r)
{
u.sum = l.sum + r.sum;
u.lmax = max(l.lmax, l.sum + r.lmax);
u.rmax = max(r.rmax, r.sum + l.rmax);
u.tmax = max(max(l.tmax, r.tmax), l.rmax + r.lmax);
}
void pushup(int u)
{
pushup(tr[u], tr[u << 1], tr[u << 1 | 1]);
}
void build(int u, int l, int r)
{
if (l == r) tr[u] = {
l, r, w[r], w[r], w[r], w[r]};
else
{
tr[u] = {
l, r};
int mid = l + r >> 1;
build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
pushup(u);
}
}
void modify(int u, int x, int v)
{
if (tr[u].l == x && tr[u].r == x) tr[u] = {
x, x, v, v, v, v};
else
{
int mid = tr[u].l + tr[u].r >> 1;
if (x <= mid) modify(u << 1, x, v);
else modify(u << 1 | 1, x, v);
pushup(u);
}
}
Node query(int u, int l, int r)
{
if (tr[u].l >= l && tr[u].r <= r) return tr[u];
else
{
int mid = tr[u].l + tr[u].r >> 1;
if (r <= mid) return query(u << 1, l, r);
else if (l > mid) return query(u << 1 | 1, l, r);
else
{
auto left = query(u << 1, l, r);
auto right = query(u << 1 | 1, l, r);
Node res;
pushup(res, left, right);
return res;
}
}
}
int main()
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i ++ ) scanf("%d", &w[i]);
build(1, 1, n);
int k, x, y;
while (m -- )
{
scanf("%d%d%d", &k, &x, &y);
if (k == 1)
{
if (x > y) swap(x, y);
printf("%d\n", query(1, x, y).tmax);
}
else modify(1, x, y);
}
return 0;
}
3.AcWing 246. 区间最大公约数
题目链接:https://www.acwing.com/problem/content/247/
题意:给你一个长度为n的数列A,以及m条指令。1.c l r d,表示把A[l],A[l+1],……,A[r]都加上d。2.Q l r,表示询问A[l],A[l+1]……A[r]的最大公约数,对于每个询问,输出答案。
思路:对于第一条指令,它是将一个连续的范围都加上一个数,很快就想到了差分,那么对于求一个范围[L,R]的最大公约数,我们可以求gcd(a[L],gcd(b[L+1],b[L+2]……b[R])),最大公约数具有结合律、分配律,(b数组是差分数组)。我们首先想一个每一个结点应该包含那些信息,首先肯定有左右儿子以及最大的公约数,但是我们还可以把差分数组包含在结构体内,就不用单独来储存了。
struct node
{
int l, r;
ll sum, d;///sum表示每个数对应的差分数组,d表示最大公约数。
}
我们再继续看查询,查询是在找最大的公约数,这里还是包含了三种情况,1.当前结点只在左边范围。2.当前节点只在右边范围。3.范围横跨当前结点的范围。所有查询该为:
node query(int u, int l, int r)
{
if (tr[u].l >= l && tr[u].r <= r) return tr[u];
else
{
int mid = tr[u].l + tr[u].r >> 1;
if (r <= mid) return query(u << 1, l, r);
else if (l > mid) return query(u << 1 | 1, l, r);
else
{
auto left = query(u << 1, l, r);
auto right = query(u << 1 | 1, l, r);
node res;
pushup(res, left, right);
return res;
}
}
}
更新函数,需要更新的是差分和以及最大公约数,最大公数等于两个数的最大公约数的公约数:
void pushup(node &u, node &l, node &r)
{
u.sum = l.sum + r.sum;
u.d = gcd(l.d, r.d);
}
void pushup(int u)
{
pushup(tr[u], tr[u << 1], tr[u << 1 | 1]);
}
在说一下修改函数,因为这里的修改是在一个范围内添加一个数,所以我们想到了差分,所以修改函数为:
void modify(int u, int x, ll v)
{
if (tr[u].l == x && tr[u].r == x) {
ll b = tr[u].sum + v; tr[u] = {
x, x, b, b}; }
else
{
int mid = tr[u].l + tr[u].r >> 1;
if (x <= mid) modify(u << 1, x, v);
else modify(u << 1 | 1, x, v);
pushup(u);
}
}
AC代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 500010;
struct node
{
int l, r;
ll sum, d;
}tr[N * 4];
int n, m;
ll w[N];
ll gcd(int a, int b)
{
return b ? gcd(b, a % b) : a;
}
void pushup(node &u, node &l, node &r)
{
u.sum = l.sum + r.sum;
u.d = gcd(l.d, r.d);
}
void pushup(int u)
{
pushup(tr[u], tr[u << 1], tr[u << 1 | 1]);
}
void build(int u, int l, int r)
{
if (l == r) {
ll b = w[r] - w[r - 1]; tr[u] = {
l, r, b, b};}
else
{
tr[u].l = l, tr[u].r = r;
int mid = l + r >> 1;
build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
pushup(u);
}
}
void modify(int u, int x, ll v)
{
if (tr[u].l == x && tr[u].r == x) {
ll b = tr[u].sum + v; tr[u] = {
x, x, b, b}; }
else
{
int mid = tr[u].l + tr[u].r >> 1;
if (x <= mid) modify(u << 1, x, v);
else modify(u << 1 | 1, x, v);
pushup(u);
}
}
node query(int u, int l, int r)
{
if (tr[u].l >= l && tr[u].r <= r) return tr[u];
else
{
int mid = tr[u].l + tr[u].r >> 1;
if (r <= mid) return query(u << 1, l, r);
else if (l > mid) return query(u << 1 | 1, l, r);
else
{
auto left = query(u << 1, l, r);
auto right = query(u << 1 | 1, l, r);
node res;
pushup(res, left, right);
return res;
}
}
}
int main()
{
cin >> n >> m;
for (int i = 1; i <= n; i++) cin >> w[i];
build(1, 1, n);
int l, r;
ll d;
char op[2];
while (m --)
{
scanf("%s%d%d", op, &l, &r);
if (*op == 'Q')
{
auto left = query(1, 1, l);
node right = {
0, 0, 0, 0};
if (l + 1 <= r) right = query(1, l + 1, r);
printf("%lld\n", abs(gcd(left.sum, right.d)));
}
else
{
scanf("%lld", &d);
///差分
modify(1, l, d);
if (r + 1 <= n) modify(1, r + 1, -d);
}
}
return 0;
}
4.AcWing 247. 亚特兰蒂斯
题目链接:https://www.acwing.com/problem/content/249/
题意:给你描绘了一个地图,地图上的图形都是矩形,有可能有重合,问你这些矩形的面积是多少。
思路:线段树+扫描线。
如图,假设是给定的图形:
那么我们变换一下,得到了应该求的面积的区域:
我们将重合的部分去掉,就得到了实际的面积区域,我们在做如下变换:
那么我们要求的面积就是图中绿色的阴影部分,那么我们现在的问题就是每一个竖条间的阴影部分的高度是多少,因为两个竖条之间的距离好找,现在的难点就是找高度。我们现在可以在纵坐标上做一个线段树,在此之前我们还需要做一个操作,就是把阴影部分的每一条竖着的边加上一个权值,看成一个带边权的有向边,如下图:
这样我们就把每条边看作了一个有权值的边。然后我们只需要在每一个竖边对应的y轴上加上它的权值,这样我们从左往右看,就可以把每个边看作一个操作。然后我们通过看每个线段内的权值,就可以知道当前区间被多少个矩形覆盖,那么我们现在可以进行两个操作:1.将某个区间[L,R] + k(k = 1或者k = -1)。2.整个区间中,长度大于0的区间总长是多少。那么我们线段树所需要维护的有:cnt,被覆盖的次数,len,不考虑祖先节点cnt的前提下,cnt>0的区间总长。现在我们要知道扫描线的几个性质:1.永远只考虑根节点的信息。2.所有的操作都是成对出现的,且先加后减。(这里就说明了可以不用pushdown)。我们使用扫描线来求面积,顾名思义,就是用一根竖直的线从左向右的扫描,但后的动态的保存面积,只有是遇到区间的cnt是大于0的就更新面积。
那么我们现在来看一下每个节点所包含的信息,左右儿子,当前区间出现的次数,以及当前区间的长度,所以一共包含了四个量:
struct node
{
int l, r;
int cnt;
double len;
}tr[N * 8];
这里在补充一下,因为题目中说明了坐标不一定是整数,那么可能就会有小数,所以这里我们需要离散化来保存坐标:
vector<double> ys;
int find(double x)
{
return lower_bound(ys.begin(), ys.end(), x) - ys.begin();
}
继续,我们上面找到了线段树上每个线段的信息,现在我们要考虑扫描线的信息了
struct Segment
{
double x, y1, y2;
int k;///判断是那个竖边
bool operator < (const Segment &t) const
{
return x < t.x;
}
}seg[N * 2];
接下来我们先看主函数,因为我们每次都是从根节点获得信息,根节点中保存了当前状态下y轴的那些区间出现了的次数,我们在跟新了答案之后还需要进行操作,因为可能遇到边权是-1的竖边,所以我们需要修改,如下:
double res = 0;
for (int i = 0; i < n * 2; i++)
{
if (i > 0) res += tr[1].len * (seg[i].x - seg[i - 1].x);
modify(1, find(seg[i].y1), find(seg[i].y2) - 1, seg[i].k);
}
然后是pushup函数,这里要分成这么几种情况:1.完全覆盖,那么此时的tr[u].cnt大于0,那么可以直接得到区间长度,2.tr[u].cnt = 0,但是不是叶子节点,那么父节点的长度就等于两个儿子节点的长度,3.是叶子节点,那么长度就等于0。所以pushup函数就是:
void pushup(int u)
{
///完全覆盖,
if (tr[u].cnt) tr[u].len = ys[tr[u].r + 1] - ys[tr[u].l];
///不完全覆盖,cnt = 0,首先判断是不是叶节点。
else if (tr[u].l != tr[u].r) tr[u].len = tr[u << 1].len + tr[u << 1 | 1].len;
else tr[u].len = 0;///叶子节点
}
AC代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
struct Segment
{
double x, y1, y2;
int k;///判断是那个竖边
bool operator < (const Segment &t) const
{
return x < t.x;
}
}seg[N * 2];
struct node
{
int l, r;
int cnt;
double len;
}tr[N * 8];
int n;
vector<double> ys;
int find(double x)
{
return lower_bound(ys.begin(), ys.end(), x) - ys.begin();
}
void pushup(int u)
{
///完全覆盖,
if (tr[u].cnt) tr[u].len = ys[tr[u].r + 1] - ys[tr[u].l];
///不完全覆盖,cnt = 0,首先判断是不是叶节点。
else if (tr[u].l != tr[u].r) tr[u].len = tr[u << 1].len + tr[u << 1 | 1].len;
else tr[u].len = 0;///叶子节点
}
void build(int u, int l, int r)
{
tr[u] = {
l, r, 0, 0};
if (l != r)
{
int mid = l + r >> 1;
build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
}
}
void modify(int u, int l, int r, int k)
{
if (tr[u].l >= l && tr[u].r <= r) {
tr[u].cnt += k; pushup(u);}
else
{
int mid = tr[u].l + tr[u].r >> 1;
if (l <= mid) modify(u << 1, l, r, k);
if (r > mid) modify(u << 1 | 1, l, r, k);
pushup(u);
}
}
int main()
{
int T = 1;
while (cin >> n, n)
{
ys.clear();
for (int i = 0, j = 0; i < n; i++)
{
double x1, y1, x2, y2; cin >> x1 >> y1 >> x2 >> y2;
seg[j ++] = {
x1, y1, y2, 1}, seg[j ++] = {
x2, y1, y2, -1};
ys.push_back(y1), ys.push_back(y2);
}
sort(ys.begin(), ys.end());
ys.erase(unique(ys.begin(), ys.end()), ys.end());
build(1, 0, ys.size() - 2);///建树,0开始,-2。
sort(seg, seg + n * 2);
double res = 0;
for (int i = 0; i < n * 2; i++)
{
if (i > 0) res += tr[1].len * (seg[i].x - seg[i - 1].x);
modify(1, find(seg[i].y1), find(seg[i].y2) - 1, seg[i].k);
}
printf("Test case #%d\n", T ++ );
printf("Total explored area: %.2lf\n\n", res);
}
return 0;
}
5.AcWing 1277. 维护序列
题目链接:https://www.acwing.com/problem/content/1279/
题意:这个跟第二题类似,只不过操作边长了三种:1.把数列中的一段数全部乘一个值;2.把数列中的一段数全部加一个值;3.询问数列中的一段数的和,由于答案可能很大,你只需输出这个数模 P 的值。
思路:这个题主要是pushdown有一些改变,我们向下维护,是要维护当前节点的两个儿子节点,然后要把ada = 0,mul = 1,然后我们在单独建一个函数来计算向下维护的节点的add、mul以及sum值,代码如下:
void eval(node &t, int add, int mul)
{
t.sum = ((ll)t.sum * mul + (ll)(t.r - t.l + 1) * add) % p;
t.mul = (ll)t.mul * mul % p;
t.add = ((ll)t.add * mul + add) % p;
}
void pushdown(int u)
{
eval(tr[u << 1], tr[u].add, tr[u].mul);
eval(tr[u << 1 | 1], tr[u].add, tr[u].mul);
tr[u].add = 0, tr[u].mul = 1;
}
AC代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
typedef long long ll;
struct node
{
int l, r;
ll sum, add, mul