题目传送门:3149: 弱水三千,只取一瓢 -- Power OJ
做完power oj 3150就过来做这个3149,这个题的意思呢,就是给你n长度区间内有n盆花,花的种类只有64种,在初始花的情况下,通过操作1,将[l,r]区间内 花的种类修改成某一种,操作2呢,就是询问区间内画的种类数。
解题思路:
比较类似的,这题也是线段树区间修改+区间查询,不同点在于这题不是26颗线段树了,而是64颗,维护的就是该区间对应植株种类的存在情况,用bool数组存,同时也需要用到懒标;记录该区间被修改成了哪个种类的植株。题目很简单,可以与3150对照着看。
AC代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=1e5+10;
#define ll (now<<1|1)
#define rr ll+1
#define mid ((l+r)>>1)
int n,q,m;
int a[N];
struct node{
bool st[65];
int f;
}tree[4*N];
void pushup(int now)
{
for (int i=1; i<=m; i++)
{
if(tree[ll].st[i]||tree[rr].st[i]) tree[now].st[i]=true;
else tree[now].st[i]=false;
}
}
void pushdown(int now)
{
if(tree[now].f)
{
int f=tree[now].f;
tree[ll].f=tree[rr].f=f;
tree[now].f=0;
memset(tree[ll].st, 0, sizeof(tree[ll].st));
memset(tree[rr].st, 0, sizeof(tree[rr].st));
tree[ll].st[f]=tree[rr].st[f]=true;
}
}
void build_tree(int now=0,int l=1,int r=n)
{
if(l==r)
{
tree[now].st[a[l]]=true;
}
else{
build_tree(ll,l,mid);
build_tree(rr,mid+1,r);
pushup(now);
}
}
void update(int idxl,int idxr,int v,int now=0,int l=1,int r=n)//修改操作
{
if(l>=idxl&&r<=idxr)
{
tree[now].f=v;
memset(tree[now].st, 0, sizeof(tree[now].st));
tree[now].st[v]=true;
return;
}
pushdown(now);
if(idxl<=mid) update(idxl, idxr, v,ll,l,mid);//左边界在mid左边就更新左节点
if(idxr>mid) update(idxl, idxr, v,rr,mid+1,r);//右边界在mid右边才更新右节点
pushup(now);
}
node query(int idxl,int idxr,int now=0,int l=1,int r=n)//查询操作
{
if(l>=idxl&&r<=idxr) return tree[now];
pushdown(now);
if(idxr<=mid) return query(idxl, idxr,ll,l,mid);//右边界在mid左边就只需查询左节点
if(idxl>mid) return query(idxl, idxr,rr,mid+1,r);//左边界在mid右边也就只需查询右节点
node aa=query(idxl, idxr,ll,l,mid);//mid在中间就只能老实查询两边咯
node bb=query(idxl, idxr,rr,mid+1,r);
for (int i=1; i<=m; i++) if(aa.st[i]||bb.st[i]) aa.st[i]=true;
return aa;
}
int main()
{
scanf("%d%d%d",&n,&q,&m);
for (int i=1; i<=n; i++) scanf("%d",&a[i]);
build_tree();
int op;
while(q--)
{
scanf("%d",&op);
if(op&1)
{
int l,r,x;
scanf("%d%d%d",&l,&r,&x);
update(l, r, x);
}else{
int l,r;
scanf("%d%d",&l,&r);
node a=query(l, r);
int cnt=0;
for (int i=1; i<=m; i++) if(a.st[i]) cnt++;
printf("%d\n",cnt);
}
}
return 0;
}