ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

线段树上状压(位运算)

2022-09-08 20:32:03  阅读:203  来源: 互联网

标签:颜色 运算 int 线段 状压 char 二进制 区间


洛谷P1558

分析:

颜色类型只有 \(30\) 种,可以利用二进制进行状压。
线段树维护一个二进制数表示区间的颜色为哪一种,将这个区间的颜色进行状压,每一种颜色对应二进制数的某一位。合并区间时将两个子节点的数按位或即可,题目区间修改为直接覆盖,统计答案时只需统计对应区间的数有多少个 \(1\) 即可。

代码:

#include <cstdio>
#include <iostream>
#define maxn 4000005
using namespace std;

int l, r, k, p, T;
char check;
int n, q, a, b, c;
inline int read() {
    int x = 0, f = 1;
    char c = getchar();
    while (c < '0' || c > '9') {
        if (c == '-')
            f = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9') x = (x << 3) + (x << 1) + (c ^ 48), c = getchar();
    return x * f;
}

struct Tree {
    int l, r, ans, xr, lazy, len;
} t[maxn << 2];

inline void update(int node) { t[node].ans = t[node << 1].ans | t[node << 1 | 1].ans; }

inline void pushdown(int node) {
    if (t[node].lazy) {
        t[node << 1].lazy = t[node << 1 | 1].lazy = t[node].lazy;
        t[node << 1].ans = t[node << 1 | 1].ans = 1 << t[node].lazy;
        t[node].lazy = 0;
    }
}

inline void build(int l, int r, int node) {
    t[node].l = l, t[node].r = r, t[node].len = r - l + 1;
    if (l == r) {
        t[node].ans = 2;
        return;
    }
    //初始类型都为1
    int mid = l + r >> 1;
    build(l, mid, node << 1);
    build(mid + 1, r, node << 1 | 1);
    update(node);
}

inline void change(int l, int r, int k, int node) {
    if (l <= t[node].l && t[node].r <= r) {
        t[node].ans = (1 << k);
        t[node].lazy = k;
        return;
    }
    //因为是染色所以直接覆盖
    int mid = t[node].l + t[node].r >> 1;
    pushdown(node);
    if (l <= mid)
        change(l, r, k, node << 1);
    if (mid < r)
        change(l, r, k, node << 1 | 1);
    update(node);
}

inline int ask(int l, int r, int node) {
    if (l <= t[node].l && t[node].r <= r) {
        return t[node].ans;
    }
    pushdown(node);
    int ans = 0;
    int mid = t[node].l + t[node].r >> 1;
    if (l <= mid)
        ans |= ask(l, r, node << 1);
    if (mid < r)
        ans |= ask(l, r, node << 1 | 1);
    return ans;
}

int main() {
    n = read(), q = read();
    T = read();
    build(1, n, 1);
    while (T--) {
        int x, y, z;
        char op;
        cin >> op >> x >> y;
        if (x > y)
            swap(x, y);
        if (op == 'C') {
            z = read();
            change(x, y, z, 1);
        } else {
            int H0ndomach1 = ask(x, y, 1);

            int ans = 0;

            for (int i = 1; i <= q; i++)
                if (H0ndomach1 & (1 << i))
                    ans++;
            //处理出有多少个1即为不同的种类数
            printf("%d\n", ans);
        }
    }
    return 0;
}

标签:颜色,运算,int,线段,状压,char,二进制,区间
来源: https://www.cnblogs.com/DAIANZE/p/16670773.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有