首页 > 代码库 > uva 11402 - Ahoy, Pirates!(线段树)
uva 11402 - Ahoy, Pirates!(线段树)
题目链接:uva 11402 - Ahoy, Pirates!
题目大意:给定给一个字符串,字符串的给定方式为各个循坏单位的循环次数和循环单位,然后是Q次操作。
- F:将l~r之间的数变成1
- E:将l~r之间的束变成0
- I:将l~r之间的数0变1,1变0
- Q:查询l~r之间1的个数
解题思路:线段树,注意pushdown函数中I操作不属于覆盖操作,要与子节点中的懒惰标记判断关系处理。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 1024005;
#define lson(x) ((x)<<1)
#define rson(x) (((x)<<1)+1)
int N;
bool v[maxn];
struct Node {
int l, r;
int vset, vmix, val;
void set (int l, int r, int val, int vset) {
this->l = l;
this->r = r;
this->val = val;
this->vset = vset;
}
}node[maxn*4];
void set_node (int u, int vset) {
if (vset == 1 || (node[u].vset == 2 && vset == 3)) {
node[u].val = node[u].r - node[u].l + 1;
node[u].vset = 1;
} else if (vset == 2 || (node[u].vset == 1 && vset == 3)) {
node[u].val = 0;
node[u].vset = 2;
} else {
node[u].val = node[u].r - node[u].l + 1 - node[u].val;
node[u].vset = vset - node[u].vset;
}
}
void pushup (int u) {
node[u].set(node[lson(u)].l, node[rson(u)].r, node[lson(u)].val + node[rson(u)].val, 0);
}
void pushdown (int u) {
if (node[u].vset) {
set_node(lson(u), node[u].vset);
set_node(rson(u), node[u].vset);
}
node[u].vset = 0;
}
void build_segTree (int u, int l, int r) {
if (l == r) {
node[u].set(l, r, v[l] ? 1 : 0, 0);
return;
}
int mid = (l + r) / 2;
build_segTree(lson(u), l, mid);
build_segTree(rson(u), mid+1, r);
pushup(u);
}
void init () {
N = 0;
memset(v, 0, sizeof(v));
int n, x;
char s[100];
scanf("%d", &n);
for (int i = 0; i < n; i++) {
scanf("%d%s", &x, s);
int len = strlen(s);
for (int j = 0; j < len; j++) {
if (s[j] == ‘1‘) {
for (int k = 0; k < x; k++)
v[k*len+j+N] = 1;
}
}
N += x * len;
}
build_segTree(1, 0, N-1);
}
void set_segTree (int u, int l, int r, int vset) {
if (l <= node[u].l && node[u].r <= r) {
set_node(u, vset);
return;
}
pushdown(u);
int mid = (node[u].l + node[u].r) / 2;
if (l <= mid)
set_segTree(lson(u), l, r, vset);
if (r > mid)
set_segTree(rson(u), l, r, vset);
pushup(u);
}
int query_segTree (int u, int l, int r) {
if (l <= node[u].l && node[u].r <= r)
return node[u].val;
pushdown(u);
int mid = (node[u].l + node[u].r) / 2;
int ret = 0;
if (l <= mid)
ret += query_segTree(lson(u), l, r);
if (r > mid)
ret += query_segTree(rson(u), l, r);
pushup(u);
return ret;
}
void solve () {
int n, cas = 1, l, r;
char s[5];
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
scanf("%s%d%d", s, &l, &r);
if (s[0] == ‘F‘)
set_segTree(1, l, r, 1);
else if (s[0] == ‘E‘)
set_segTree(1, l, r, 2);
else if (s[0] == ‘I‘)
set_segTree(1, l, r, 3);
else
printf("Q%d: %d\n", cas++, query_segTree(1, l, r));
}
}
int main () {
int cas;
scanf("%d", &cas);
for (int kcas = 1; kcas <= cas; kcas++) {
init();
printf("Case %d:\n", kcas);
solve();
}
return 0;
}
uva 11402 - Ahoy, Pirates!(线段树)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。