欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

P2184 贪婪大陆

程序员文章站 2022-04-20 19:16:04
题目 "P2184 贪婪大陆" 解析 线段树,差分? 在所修改的区间的开头位置+1,表示从这个位置开始往后开始埋一种地雷,在结尾位置+1,表示在这个位置有一种地雷被埋完 查询的时候我们就只需要查询 $[1,r]$中开头的位置,表示$1$到r*埋了多少种类型的地雷 $[1,l 1]$中结尾的个数,表 ......

题目

p2184 贪婪大陆

解析

线段树,差分?
在所修改的区间的开头位置+1,表示从这个位置开始往后开始埋一种地雷,在结尾位置+1,表示在这个位置有一种地雷被埋完
查询的时候我们就只需要查询

  • \([1,r]\)中开头的位置,表示\(1\)到r*埋了多少种类型的地雷
  • \([1,l-1]\)中结尾的个数,表示\(1\)\(l-1\)中有多少种类型的地雷被埋完,因为在\(l\)之前就埋完了,所以不会影响\([l,r]\)内的答案

所以查询时输出前者减后者的差就可以了。
线段树维护单点修改+区间查询

代码

#include <bits/stdc++.h>
using namespace std;
const int n = 1e6 + 10;
int n, m;
int d[n];
class tree {
  public:
    int len, sta, end;
} t[n];

#define lson rt << 1
#define rson rt << 1 | 1

template<class t>inline void read(t &x) {
    x = 0; int f = 0; char ch = getchar();
    while (!isdigit(ch)) f |= (ch == '-'), ch = getchar();
    while (isdigit(ch)) x = x * 10 + ch - '0', ch = getchar();
    x = f ? -x : x;
    return;
}

void build(int l, int r, int rt) {
    t[rt].len = r - l + 1;
    if (l == r) return;
    int m = (l + r) >> 1;
    build(l, m, lson), build(m + 1, r, rson);
}

void update_sta(int l, int c, int l = 1, int r = n, int rt = 1) {
    if (l == r) {
        t[rt].sta += c;
        return ;
    }
    int m = (l + r) >> 1;
    if (l <= m) update_sta(l, c, l, m, lson);
    else update_sta(l, c, m + 1, r, rson);
    t[rt].sta = t[lson].sta + t[rson].sta;
}

void update_end(int l, int c, int l = 1, int r = n, int rt = 1) {
    if (l == r) {
        t[rt].end += c;
        return ;
    }
    int m = (l + r) >> 1;
    if (l <= m) update_end(l, c, l, m, lson);
    else update_end(l, c, m + 1, r, rson);
    t[rt].end = t[lson].end + t[rson].end;
}

int query_sta(int l, int r, int l = 1, int r = n, int rt = 1) {
    if (l <= l && r <= r) return t[rt].sta;
    int m = (l + r) >> 1, ans = 0;
    if (l <= m) ans += query_sta(l, r, l, m, lson);
    if (r > m) ans += query_sta(l, r, m + 1, r, rson);
    return ans;
}

int query_end(int l, int r, int l = 1, int r = n, int rt = 1) {
    if (l <= l && r <= r) return t[rt].end;
    int m = (l + r) >> 1, ans = 0;
    if (l <= m) ans += query_end(l, r, l, m, lson);
    if (r > m) ans += query_end(l, r, m + 1, r, rson);
    return ans;
}

int main() {
    read(n), read(m);
    build(1, n, 1);
    for (int i = 1, x, y, opt; i <= m; ++i) {
        read(opt) , read(x), read(y);
        if (opt == 1) update_sta(x, 1), update_end(y, 1);
        else printf("%d\n", query_sta(1, y) - query_end(1, x - 1));
    }
}