线段树
线段树是一个二叉树,每个结点保存数组nums在区间 [s,e]的最小值、最大值或者总和等信息。线段树可以用树也可以用数组(堆式存储)来实现。对于数组实现,假设根结点的下标为 1,如果一个结点在数组的下标为node,那么它的左子结点下标为 node×2,右子结点下标为node×2+1,线段树可以在 O(logN) 的时间复杂度内实现单点修改、区间修改、区间查询(区间求和,求区间最大值,求区间最小值)等操作。
构建线段树
线段树将每个长度不为 1 的区间划分成左右两个区间递归求解,把整个线段划分为一个树形结构,通过合并左右两区间信息来求得该区间的信息。这种数据结构可以方便的进行大部分的区间操作。有个大小为 5 的数组 a={10,11,12,13,14},要将其转化为线段树,有以下做法:设线段树的根节点编号为 1,用数组 d 来保存我们的线段树,di 用来保存线段树上编号为 i 的节点的值(这里每个节点所维护的值就是这个节点所表示的区间总和)。
如果要求修改区间 [l,r],把所有包含在区间 [l,r] 中的节点都遍历一次、修改一次,时间复杂度无法承受。我们这里要引入一个叫做 「懒惰标记」 的东西。懒惰标记,简单来说,就是通过延迟对节点信息的更改,从而减少可能不必要的操作次数。每次执行修改时,我们通过打标记的方法表明该节点对应的区间在某一次操作中被更改,但不更新该节点的子节点的信息。实质性的修改则在下一次访问带有标记的节点时才进行。
class Solution {
public long[] handleQuery(int[] nums1, int[] nums2, int[][] queries) {
int n = nums1.length;
int m = queries.length;
SegTree tree = new SegTree(nums1);
long sum = 0;
for (int num : nums2) {
sum += num;
}
List<Long> list = new ArrayList<Long>();
for (int i = 0; i < m; i++) {
if (queries[i][0] == 1) {
int l = queries[i][1];
int r = queries[i][2];
tree.reverseRange(l, r);
} else if (queries[i][0] == 2) {
sum += (long) tree.sumRange(0, n - 1) * queries[i][1];
} else if (queries[i][0] == 3) {
list.add(sum);
}
}
long[] ans = new long[list.size()];
for (int i = 0; i < list.size(); i++) {
ans[i] = list.get(i);
}
return ans;
}
}
class SegTree {
private SegNode[] arr;
public SegTree(int[] nums) {
int n = nums.length;
arr = new SegNode[n * 4 + 1];
build(1, 0, n - 1, nums);
}
public int sumRange(int left, int right) {
return query(1, left, right);
}
public void reverseRange(int left, int right) {
modify(1, left, right);
}
public void build(int id, int l, int r, int[] nums) {
arr[id] = new SegNode();
arr[id].r = r;
arr[id].l = l;
arr[id].lazytag = false;
if (l == r) {
arr[id].sum = nums[l];
return;
}
int mid = (l + r) >> 1;
build(2 * id, l, mid, nums);
build(2 * id +1, mid + 1, r, nums);
arr[id].sum = arr[2 * id].sum + arr[2 * id + 1].sum;
}
/* pushdown函数:下传懒标记,即将当前区间的修改情况下传到其左右孩子结点 */
public void pushdown(int x) {
if (arr[x].lazytag) {
arr[2 * x].lazytag = !arr[x * 2].lazytag;
arr[2 * x].sum = arr[2 * x].r - arr[x * 2].l + 1 - arr[2 * x].sum;
arr[2 * x + 1].lazytag = !arr[2 * x + 1].lazytag;
arr[2 * x + 1].sum = arr[2 * x + 1].r - arr[2 * x + 1].l + 1 - arr[2 * x + 1].sum;
arr[x].lazytag = false;
}
}
/* 区间修改 */
public void modify(int id, int l, int r) {
if (arr[id].l >= l && arr[id].r <= r) {
arr[id].sum = (arr[id].r - arr[id].l + 1) - arr[id].sum;
arr[id].lazytag = !arr[id].lazytag;
return;
}
pushdown(id);
int mid = (arr[id].l + arr[id]. r) >> 1;
if (arr[2 * id].r >= l) {
modify(2 * id, l, r);
}
if(arr[2 * id + 1].l <= r) {
modify(2 * id + 1, l, r);
}
arr[id].sum = arr[2 * id].sum + arr[2 * id + 1].sum;
}
/* 区间查询 */
public int query(int id, int l, int r) {
if (arr[id].l >= l && arr[id].r <= r) {
return arr[id].sum;
}
if (arr[id].r < l || arr[id].l > r) {
return 0;
}
pushdown(id);
int res = 0;
if (arr[2 * id].r >= l) {
res += query(2 * id, l, r);
}
if (arr[2 * id + 1].l <= r) {
res += query(2 * id + 1, l, r);
}
return res;
}
}
class SegNode {
public int l, r, sum;
public boolean lazytag;
public SegNode() {
this.l = 0;
this.r = 0;
this.sum = 0;
this.lazytag = false;
}
}