1. 程式人生 > >HDU 3397 Sequence operation(區間合並 + 區間更新)

HDU 3397 Sequence operation(區間合並 + 區間更新)

track define truct mat const 區間合並 http build scrip

題目鏈接:

pid=3397">http://acm.hdu.edu.cn/showproblem.php?pid=3397

題意:給定n個數,由0,1構成。共同擁有5種操作。

每一個操作輸入3個數,op,a。b。

  1. op == 0。將區間[a,b]賦值為0。
  2. op == 1,將區間[a,b]賦值為1;
  3. op == 2。將區間[a。b]內的01反轉;
  4. op == 3。查詢區間[a。b]中1的個數。
  5. op == 4,查詢區間[a。b]中連續1的最大長度。

思路:區間合並 + 區間更新。

每一個結點存7個數:

  1. 區間內1的個數c1。
  2. 從區間左端點開始連續1的最大長度l1。
  3. 以區間右端點結束連續1的最大長度r1。
  4. 區間內連續1的最大長度m1。

  5. 從區間左端點開始連續0的最大長度l0。
  6. 以區間右端點結束連續0的最大長度r0。
  7. 區間內連續0的最大長度m0。

記錄0的連續是為了反轉後統計結果方便。

代碼:

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#include <string>

using namespace std;

#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
const int N = 1e5 + 10; const int INF = 0x7f7f7f7f; struct Node { int c1; int l1, r1, m1; int l0, r0, m0; Node() {} Node(int a, int b, int c, int d, int e, int f, int g) { c1 = a; l1 = b; r1 = c; m1 = d; l0 = e; r0 = f; m0 = g; } void
reversal(int s) { c1 = s; swap(l1, l0); swap(r1, r0); swap(m1, m0); } }; Node node[N << 2]; int cover[N << 2]; int XOR[N << 2]; void pushup(int rt, int l, int r) { int m = (l + r) >> 1; int lenl = m - l + 1; int lenr = r - m; node[rt].c1 = node[rt << 1].c1 + node[rt << 1 | 1].c1; if (node[rt << 1].l1 == lenl) node[rt].l1 = node[rt << 1].l1 + node[rt << 1 | 1].l1; else node[rt].l1 = node[rt << 1].l1; if (node[rt << 1 | 1].r1 == lenr) node[rt].r1 = node[rt << 1 | 1].r1 + node[rt << 1].r1; else node[rt].r1 = node[rt << 1 | 1].r1; if (node[rt << 1].l0 == lenl) node[rt].l0 = node[rt << 1].l0 + node[rt << 1 | 1].l0; else node[rt].l0 = node[rt << 1].l0; if (node[rt << 1 | 1].r0 == lenr) node[rt].r0 = node[rt << 1 | 1].r0 + node[rt << 1].r0; else node[rt].r0 = node[rt << 1 | 1].r0; node[rt].m1 = max(node[rt << 1].r1 + node[rt << 1 | 1].l1, max(node[rt << 1].m1, node[rt << 1 | 1].m1)); node[rt].m0 = max(node[rt << 1].r0 + node[rt << 1 | 1].l0, max(node[rt << 1].m0, node[rt << 1 | 1].m0)); } void pushdown(int rt, int l, int r) { int m = (l + r) >> 1; int lenl = m - l + 1; int lenr = r - m; if (cover[rt] != -1) { if (cover[rt] == 1) { node[rt << 1] = Node(lenl, lenl, lenl, lenl, 0, 0, 0); node[rt << 1 | 1] = Node(lenr, lenr, lenr, lenr, 0, 0, 0); } else { node[rt << 1] = Node(0, 0, 0, 0, lenl, lenl, lenl); node[rt << 1 | 1] = Node(0, 0, 0, 0, lenr, lenr, lenr); } cover[rt << 1] = cover[rt << 1 | 1] = cover[rt]; XOR[rt << 1] = XOR[rt << 1 | 1] = 0; cover[rt] = -1; } if (XOR[rt] != 0) { node[rt << 1].reversal(lenl - node[rt << 1].c1); node[rt << 1 | 1].reversal(lenr - node[rt << 1 | 1].c1); if (cover[rt << 1] != -1) cover[rt << 1] = 1 - cover[rt << 1]; else XOR[rt << 1] = 1 - XOR[rt << 1]; if (cover[rt << 1 | 1] != -1) cover[rt << 1 | 1] = 1 - cover[rt << 1 | 1]; else XOR[rt << 1 | 1] = 1 - XOR[rt << 1 | 1]; XOR[rt] = 0; } } void build(int l, int r, int rt) { cover[rt] = -1; XOR[rt] = 0; if (l == r) { int x; scanf("%d", &x); node[rt] = Node(x, x, x, x, 1 - x, 1 - x, 1 - x); return ; } int m = (l + r) >> 1; build(lson); build(rson); pushup(rt, l, r); } void update(int op, int L, int R, int l, int r, int rt) { if (L <= l && r <= R) { if (op == 2) { node[rt].reversal(r - l + 1 - node[rt].c1); if (cover[rt] != -1) cover[rt] = 1 - cover[rt]; else XOR[rt] = 1 - XOR[rt]; } else { node[rt] = (op == 0 ? Node(0, 0, 0, 0, r - l + 1, r - l + 1, r - l + 1) : Node(r - l + 1, r - l + 1, r - l + 1, r - l + 1, 0, 0, 0)); cover[rt] = op; XOR[rt] = 0; } return ; } pushdown(rt, l, r); int m = (l + r) >> 1; if (L <= m) update(op, L, R, lson); if (R > m) update(op, L, R, rson); pushup(rt, l, r); } Node query(int op, int L, int R, int l, int r, int rt) { if (L <= l && r <= R) { return node[rt]; } pushdown(rt, l, r); int m = (l + r) >> 1; Node ql = Node(0, 0, 0, 0, 0, 0, 0); Node qr = Node(0, 0, 0, 0, 0, 0, 0); if (L <= m) ql = query(op, L, R, lson); if (R > m) qr = query(op, L, R, rson); pushup(rt, l, r); if (op == 3) { return Node(ql.c1 + qr.c1, 0, 0, 0, 0, 0, 0); } else { Node res = Node(0, 0, 0, 0, 0, 0, 0); res.m1 = max(ql.r1 + qr.l1, max(ql.m1, qr.m1)); if (ql.l1 == m - l + 1) res.l1 = ql.l1 + qr.l1; else res.l1 = ql.l1; if (qr.r1 == r - m) res.r1 = qr.r1 + ql.r1; else res.r1 = qr.r1; return res; } } int main() { int t_case; scanf("%d", &t_case); for (int i_case = 1; i_case <= t_case; i_case++) { int n, m; scanf("%d%d", &n, &m); build(1, n, 1); for (int i = 0; i < m; i++) { int op, a, b; scanf("%d%d%d", &op, &a, &b); a++; b++; if (op <= 2) { update(op, a, b, 1, n, 1); } else { Node res = query(op, a, b, 1, n, 1); if (op == 3) printf("%d\n", res.c1); else printf("%d\n", res.m1); } } } return 0; }

HDU 3397 Sequence operation(區間合並 + 區間更新)