区间染色/线段树（ZOJ - 1610 Count the Colors ）

最新推荐文章于 2020-03-16 12:32:17 发布

原创最新推荐文章于 2020-03-16 12:32:17 发布 · 309 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#线段树

数据结构专栏收录该内容

53 篇文章

订阅专栏

本文介绍了一种区间更新和查询的优化方法，通过将区间右端点减一并使用线段树进行更新，解决了区间覆盖问题。同时，文章分享了AC代码实现，包括线段树的建立、更新和查询操作。

题目链接
模板题，类似题详解
注意的是，题目所给的是区间，不是点，5 6 3并不是给5、6两个点染3号色，而是在(5, 6)上。~~要不是1号样例比较良心，我可以wa一天~~
在update时，让右端点减一，就可以应对这种情况（即，把每个区间用它的左端点表示）
下面是ac代码：

#include <iostream>
#include <string>
#include <cstring>
#include <queue>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#define ll long long
using namespace std;
const int N = 100005;
struct Node
{
    int l, r;
    int gg;
}tr[(N<<2)+5];
int cnt[N];
int ans[N];
int mx = 0;
void build(int p, int l, int r)
{
    tr[p].l = l; tr[p].r = r;
    if (l == r){tr[p].gg = -1; return;}
    int mid = (l + r) >> 1;
    build(p<<1, l, mid);
    build(p<<1|1, mid+1, r);
    tr[p].gg = -1;
}
void spread(int q)
{
    if (tr[q].gg == -1) return;
    tr[q<<1].gg = tr[q].gg;
    tr[q<<1|1].gg = tr[q].gg;
    tr[q].gg = -1;
}
void update(int q, int l, int r, int v)
{
    if (l <= tr[q].l && r >= tr[q].r)
    {tr[q].gg = v; return;}
    spread(q);
    int mid = (tr[q].l + tr[q].r) >> 1;
    if (l <= mid) update(q<<1, l, r, v);
    if (r  > mid) update(q<<1|1, l, r, v);
}
int ask(int p, int k)
{
    if (tr[p].l == tr[p].r) return tr[p].gg;
    spread(p);
    int mid = (tr[p].l + tr[p].r) >> 1;
    if (k <= mid) return ask(p<<1, k);
    else return ask(p<<1|1, k);
}
int main()
{
    int q;
    while(scanf("%d", &q) != EOF)
    {
        build(1, 0, 8005);
        while(q--)
        {
            int x, y, w;
            scanf("%d%d%d", &x, &y, &w);
            update(1, x, y-1, w);
        }
        memset(cnt, -1, sizeof(cnt));
        for (int i = 0; i <= 8005; i++)
            cnt[i] = ask(1, i);
     //   for (int i = 0; i<= 10; i++) cout << cnt[i] <<" ";
       // cout <<endl;
        int m = unique(cnt, cnt + 8006) - cnt;
     //   for (int i = 0; i<= 10; i++) cout << cnt[i] <<" ";
     //   cout <<endl;
        memset(ans, 0, sizeof(ans));
        for (int i = 0; i < m; i++)
            if (cnt[i] != -1) ans[cnt[i]]++;
        for (int i = 0; i < 8005; i++)
            if (ans[i]) printf("%d %d\n", i, ans[i]);
        printf("\n");
    }
    return 0;
}