ZOJ-1610 线段树+两种查询方法(弥补我线段树区间填充的短板)

ZOJ-1610 线段树+两种查询方法(弥补我线段树区间填充的短板)

题意

题意:给一个n,代表n次操作,接下来每次操作表示把[l,r]区间的线段涂成k的颜色其中,l,r,k的范围都是0到8000

这题比较坑的是不能拿n建树,不然就会segmentation fault,我也是错了好多遍看的博客才知道,必须拿8000建树,也就是树是固定的。

解题思路一

线段树来维护点的颜色

单点查询,从左到右一个一个点的查询,这个有点暴力,但是因为每次查询都是\(logn\)也不是很大,跑了一下是60ms

代码实现一

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=8e3+7;
struct node{
int l, r;
int lazy;
}a[maxn<<2];
int ans[8007];
void down(int rt)
{
int l=rt<<1, r=rt<<1|1; a[l].lazy=a[rt].lazy; a[r].lazy=a[rt].lazy; a[rt].lazy=-1;
}
void build(int rt, int l, int r)
{
a[rt].l=l;
a[rt].r=r;
a[rt].lazy=-1; //表示当前点的范围颜色不统一
if(l==r) return ;
int mid=(l+r)>>1;
build(rt<<1, l, mid);
build(rt<<1|1, mid+1, r);
}
void update(int rt, int L, int R, int x)
{
if(L<=a[rt].l && a[rt].r<=R)
{
a[rt].lazy=x;
return ;
}
int mid=(a[rt].l+a[rt].r)>>1;
if(a[rt].lazy!=-1)
down(rt);
if(L<=mid) update(rt<<1, L, R, x);
if(R>mid) update(rt<<1|1, L, R, x);
}
int query(int rt, int x) //单点查询
{
if(a[rt].l==a[rt].r)
{
return a[rt].lazy;
}
if(a[rt].lazy!=-1) //记得一定要down一下,因为是单点查询,如果不down,这个点可能还是上一次的状态
down(rt);
int mid=(a[rt].l+a[rt].r)>>1;
if(x<=mid) return query(rt<<1, x);
else return query(rt<<1|1, x);
}
int main()
{
int n, a, b, c;
while(scanf("%d", &n)!=EOF)
{
build(1, 1, 8000);
memset(ans, 0, sizeof(ans));
for(int i=1; i<=n; i++)
{
scanf("%d%d%d", &a, &b, &c);
update(1, a+1, b, c);
}
int p1, p2;
p1=query(1, 1);
if(p1!=-2 && p1!=-1) ans[p1]++;
for(int i=2; i<=8000; i++)
{
p2=query(1, i);
if(p1!=p2){
ans[p2]++;
}
p1=p2;
}
for(int i=0; i<=8000; i++)
{
if(ans[i]!=0)
{
printf("%d %d\n", i, ans[i]);
}
}
printf("\n");
}
return 0;
}

解题思路二

类似区间查询,如果某一个点的lazy标记不是-1,那个这个区间就都是一种颜色,就不用往下递归了

如果lazy是-2,那个这个区域没有颜色,相当于中间隔开了一块区域。

查询是从左到右进行的,pre表示先前点左边的颜色,需要判断当前点与左边的点的颜色是否相同。

这个代码跑的是50ms。

代码实现二

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=8e3+7;
struct node{
int l, r;
int lazy;
}a[maxn<<2];
int ans[8007];
int pre; //pre表示先前块的颜色
void down(int rt)
{
int l=rt<<1, r=rt<<1|1; a[l].lazy=a[rt].lazy; a[r].lazy=a[rt].lazy; a[rt].lazy=-1;
}
void build(int rt, int l, int r)
{
a[rt].l=l;
a[rt].r=r;
a[rt].lazy=-2; //表示没有颜色
if(l==r) return ;
int mid=(l+r)>>1;
build(rt<<1, l, mid);
build(rt<<1|1, mid+1, r);
}
void update(int rt, int L, int R, int x)
{
if(L<=a[rt].l && a[rt].r<=R)
{
a[rt].lazy=x;
return ;
}
int mid=(a[rt].l+a[rt].r)>>1;
if(a[rt].lazy!=-1 && a[rt].lazy!=-2)
down(rt);
if(L<=mid) update(rt<<1, L, R, x);
if(R>mid) update(rt<<1|1, L, R, x);
if(a[rt<<1].lazy==a[rt<<1|1].lazy)//相当于up 看下属左右两点的颜色是否相同
{
a[rt].lazy=a[rt<<1].lazy; //相同的话,这个点就可以代表下面的两个点
}
else a[rt].lazy=-1;//否则-1表示下属两个点的颜色不统一,需要以后特殊标记
}
void query(int rt)//先当于区间查询,从左到右
{
if(a[rt].lazy==-2) //说明这个点控制的区域没有刷颜色,那么它右边的点也可以和左边的点有相同颜色
{
pre=-1;//这个标记-1
return ; //结束
}
if(a[rt].lazy!=-1)
{
if(a[rt].lazy!=pre)
{
ans[a[rt].lazy]++;
pre=a[rt].lazy;
}
return ;
}
query(rt<<1);
query(rt<<1|1);
}
int main()
{
int n, a, b, c;
while(scanf("%d", &n)!=EOF)
{
build(1, 1, 8000);
memset(ans, 0, sizeof(ans));
for(int i=1; i<=n; i++)
{
scanf("%d%d%d", &a, &b, &c);
update(1, a+1, b, c);
}
pre=-2;
query(1);
for(int i=0; i<=8000; i++)
{
if(ans[i]!=0)
{
printf("%d %d\n", i, ans[i]);
}
}
printf("\n");
}
return 0;
}
上一篇:解决EXC_BAD_ACCESS错误的一种方法--NSZombieEnabled


下一篇:如何在MacBook M1上无缝使用Win11和MacOS?