2021hdu多校4 Lawn of the Dead（线段树+模拟）

题目链接

题意：

• 对于n*m的网格，给定k个障碍，每次只能向右或向下移动，问从(1,1)开始，能到达的格子的数目。

代码：

//线段树+模拟 
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
vector<int> vv[100010];
struct node{
	int sum[100010<<2],lazy[100010<<2];  
	
	void pushup(int p){
		sum[p]=sum[p<<1]+sum[p<<1|1];
	}
	
	void build(int p,int l,int r){
		lazy[p]=-1; //因为0，1表示区间信息，故将懒标记初始化为-1 
		if(l==r){ sum[p]=1; return ;}
		int mid=l+r>>1;
		build(p<<1,l,mid);
		build(p<<1|1,mid+1,r);
		pushup(p);
	}
	
	void pushdown(int p,int tot){
		if(lazy[p]==-1) return ;
		sum[p<<1]=(tot-tot/2)*lazy[p];
		sum[p<<1|1]=(tot/2)*lazy[p];
		lazy[p<<1]=lazy[p<<1|1]=lazy[p];
		lazy[p]=-1;  //清空的lazy标记不为0，应变为-1 
	}
	//区间修改 
	void change(int p,int l,int r,int x,int y,int k){
		if(x<=l&&r<=y){ sum[p]=(r-l+1)*k; lazy[p]=k; return ;}
		pushdown(p,r-l+1);
		int mid=l+r>>1;
		if(x<=mid) change(p<<1,l,mid,x,y,k);
		if(y>mid) change(p<<1|1,mid+1,r,x,y,k);
		pushup(p);
	}
	//区间查询 
	int find2(int p,int l,int r,int x,int y){
		if(x<=l&&r<=y) return sum[p];
		pushdown(p,r-l+1);
		int mid=l+r>>1,ans=0;
		if(x<=mid) ans+=find2(p<<1,l,mid,x,y);
		if(y>mid) ans+=find2(p<<1|1,mid+1,r,x,y);
		return ans;
	}
	//找区间[x,y]中第一个1的位置 
	int find(int p,int l,int r,int x,int y){
		if(find2(1,1,y,x,y)==0) return -1;
		if(l==r) return l;
		pushdown(p,r-l+1);
		int mid=l+r>>1;
		if(x<=mid&&find2(1,1,y,x,mid)) return find(p<<1,l,mid,x,y);
		else return find(p<<1|1,mid+1,r,x,y); 
	}
}tree[2];
int main()
{
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	int _;
	cin>>_;
	while(_--){
		int n,m,k,x,y;
		cin>>n>>m>>k;
		m++;
		for(int i=1;i<=n;i++){
			vv[i].clear(); vv[i].push_back(1); //有些行没有地雷，但会被上一行影响，需人为在最前面加一个地雷 
		}
		for(int i=1;i<=k;i++){
			cin>>x>>y;
			y++; //因为加了一个雷，但线段树区间修改从1开始，可以增加一列（1到y+1) 
			vv[x].push_back(y);
		}
		tree[0].build(1,1,m);
		tree[1].build(1,1,m);
		ll ans=0;
		//第一行单独处理 
		sort(vv[1].begin(),vv[1].end());
		if(vv[1].size()>1) tree[0].change(1,1,m,vv[1][1],m,0);
		ans+=tree[0].sum[1]-1; //第一行只判断一个雷，需减去左上角的雷 
		
		int now=1;
		for(int i=2;i<=n;i++){
			int p;
			for(auto y:vv[i]){
				//对于每一个雷，找他右上角开始第一个不是雷的位置 
				if(y==m) p=m+1;
				else p=tree[now^1].find(1,1,m,y+1,m);
				//修改本行信息 
				if(p==-1) tree[now].change(1,1,m,y,m,0);
				else tree[now].change(1,1,m,y,p-1,0);
			}
			ans+=tree[now].sum[1];
	        //初始化上一行的信息 
			tree[now^1].change(1,1,m,1,m,1);
			now^=1;
		}
		cout<<ans<<"\n"; 
	}
	return 0;
}