hdu 5068 Harry And Math Teacher 线段树

今天状态不错把这题A了,那天打BC的时候看错题了= =

题意:有n层楼,每层楼有两个门,每个门都通过两个楼梯和下一层的两个门相连,m次操作,把某条楼梯的状态改变(可用与不可用),或者询问从一层到另一层有多少种走法,需要注意的就是如果你从x-1层走到了x层的a门,那么你只能从a门再走到下一层。当时就死在这了。。

一个线段树t[x][md],md表示四种从l层走到r层的进与出的四种搭配(0-0,0-1,1-0,1-1),建树、更新的时候都是分别更新这四个值,查询的时候返回一个结构,里面存的是这四个值x1,x2,x3,x4,如果L和R在左右子树都有一部分的话就用返回的两个结构相乘得到答案。

#define _CRT_SECURE_NO_WARNINGS
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define mod 1000000007
#define lson l,m,x<<1
#define rson m,r,x<<1|1
#define LL __int64
int n;
LL t[200005][4];
int dir[4][2]={0,0,0,1,1,0,1,1};
typedef struct aa
{
	LL x1,x2,x3,x4;
	aa(){};
	aa(LL a,LL b,LL c,LL d)
	{
		x1=a;x2=b;x3=c;x4=d;
	}
} Ans;
void es_tree(int l,int r,int x)
{
	if(r==l+1)
	{
		t[x][0]=t[x][1]=t[x][2]=t[x][3]=1;
		return;
	}
	int m=(l+r)/2;
	es_tree(lson);
	es_tree(rson);
	t[x][0]=t[x<<1][0]*t[x<<1|1][0]+t[x<<1][1]*t[x<<1|1][2];
	t[x][1]=t[x<<1][0]*t[x<<1|1][1]+t[x<<1][1]*t[x<<1|1][3];
	t[x][2]=t[x<<1][2]*t[x<<1|1][0]+t[x<<1][3]*t[x<<1|1][2];
	t[x][3]=t[x<<1][2]*t[x<<1|1][1]+t[x<<1][3]*t[x<<1|1][3];
	t[x][0]%=mod;
	t[x][1]%=mod;
	t[x][2]%=mod;
	t[x][3]%=mod;
}
void update(int L,int mode,int l,int r,int x)
{
	if(l==L&&r==L+1)
	{
		t[x][mode]^=1;
		return;
	}
	int m=(l+r)/2;
	if(L<m) update(L,mode,lson);
	else update(L,mode,rson);
	t[x][0]=t[x<<1][0]*t[x<<1|1][0]+t[x<<1][1]*t[x<<1|1][2];
	t[x][1]=t[x<<1][0]*t[x<<1|1][1]+t[x<<1][1]*t[x<<1|1][3];
	t[x][2]=t[x<<1][2]*t[x<<1|1][0]+t[x<<1][3]*t[x<<1|1][2];
	t[x][3]=t[x<<1][2]*t[x<<1|1][1]+t[x<<1][3]*t[x<<1|1][3];
	t[x][0]%=mod;
	t[x][1]%=mod;
	t[x][2]%=mod;
	t[x][3]%=mod;
}
Ans query(int L,int R,int l,int r,int x)
{
	if(L==l&&R==r)
	{
		return Ans(t[x][0],t[x][1],t[x][2],t[x][3]);
	}
	int m=(l+r)/2;
	if(R<=m) return query(L,R,lson);
	if(L>=m) return query(L,R,rson);
	Ans a=query(L,m,lson),b=query(m,R,rson);
	return Ans((a.x1*b.x1+a.x2*b.x3)%mod,(a.x1*b.x2+a.x2*b.x4)%mod,(a.x3*b.x1+a.x4*b.x3)%mod,(a.x3*b.x2+a.x4*b.x4)%mod);
}
int select(int from,int to)
{
	for(int i=0;i<=3;i++) 
		if(from==dir[i][0]&&to==dir[i][1]) return i;
}
int main()
{
	int md,x,f,to,a,m;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		memset(t,0,sizeof(t));
		es_tree(1,n,1);
		while(m--)
		{
			scanf("%d",&md);
			if(md)
			{
				scanf("%d%d%d",&x,&f,&to);
				int a=select(f-1,to-1);
				update(x,a,1,n,1);
			}
			else
			{
				scanf("%d%d",&f,&to);
				Ans ret=query(f,to,1,n,1);
				printf("%I64d\n",(ret.x1+ret.x2+ret.x3+ret.x4)%mod);
			}
		}
	}
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值