题目大意
在抗日战争期间,地道战在华北平原得到广泛的实施,一般而言,村庄通过一些隧道在一条线上连接,除了两端剩下的每个村庄都有两个相连。侵略者会频繁的对这些村庄进行扫荡,并且摧他们的地道,当然八路军会把这一些已经被摧毁的村庄修复的,会优先修复最近被破坏的村庄。输入一个村庄编号(1-n),求与该村庄连接的村庄个数(包括自己)。
思路
- 用一个栈模拟村庄的破坏和修复
- 利用线段树 求出每个线段左端点长度lsum(自左端点向右的连续长度),右端点长度rsum(自右端点向左连续长度) 以及区间内最长连续长度(以为可能存在一个连续长度既不包含左端点也不包含右端点)
- 破坏和修复用同一个函数,1表示存在,0表示被破坏
- 查询具体看代码
AC代码
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
const int maxn = 50005;
struct node {
int L, R, lsum, rsum, sum;
int Mid() {return (L+R)/2;}
int Len() {return (R-L+1);}
}a[maxn*4];
void pushup(int p) {
a[p].lsum = a[p << 1].lsum;
a[p].rsum = a[p << 1|1].rsum;
if (a[p << 1].lsum == a[p << 1].Len())
a[p].lsum = a[p << 1].lsum+a[p << 1|1].lsum;
if (a[p << 1|1].rsum == a[p << 1|1].Len())
a[p].rsum = a[p << 1|1].rsum+a[p << 1].rsum;
a[p].sum = max(a[p].lsum, max(a[p].rsum, a[p << 1].rsum+a[p << 1|1].lsum));
}
void Build(int p, int l, int r) {
a[p].L = l;
a[p].R = r;
a[p].lsum = a[p].rsum = a[p].sum = a[p].Len();
if (l == r) return;
Build(p << 1, l, a[p].Mid());
Build(p << 1|1, a[p].Mid()+1, r);
}
void Insert(int p, int k, int x) {
if (a[p].L == a[p].R) {
a[p].lsum = a[p].rsum = a[p].sum = x;
return;
}
if (k <= a[p].Mid())
Insert(p << 1, k, x);
else
Insert(p << 1|1, k, x);
pushup(p);
}
int Query(int p, int k) {
if (a[p].sum == 0) return 0;
if (k < a[p].L+a[p].lsum) return a[p].lsum;
if (k > a[p].R-a[p].rsum) return a[p].rsum;
if (k > a[p << 1].R-a[p << 1].rsum&&k < a[p << 1|1].L+a[p << 1|1].lsum)
return a[p << 1].rsum+a[p << 1|1].lsum;
if (k <= a[p].Mid())
return Query(p << 1, k);
else
return Query(p << 1|1, k);
}
int main() {
int n, m;
while (~scanf("%d%d", &n, &m)) {
int x;
char s[10];
stack <int> sta;
Build(1, 1, n);
while (m--) {
scanf("%s", s);
if (s[0] == 'D') {
scanf("%d", &x);
Insert(1, x, 0);
sta.push(x);
} else if (s[0] == 'R'&&sta.size()) {
Insert(1, sta.top(), 1);
sta.pop();
} else if (s[0] == 'Q') {
scanf("%d", &x);
printf("%d\n", Query(1, x));
}
}
}
return 0;
}
图片模拟过程
我一开始也不会这道题,看了别人的代码也懵,自己画图模拟了一遍就理解了,同样迷惑的,可以自行模拟一遍,大佬自动忽略~(dl应该也不会点进来)
下图模拟了建树 破坏 修复的过程,这几个搞明白了查询结合代码也很清晰了