HDU1540 Tunnel Warfare
参考博客:https://blog.csdn.net/MIKASA3/article/details/70197099
?(巨佬i了
题目大意:给出M个城市在按直线排列,用道路连接起来,现在可以把任意的城市炸裂或者修复,问一个城市连接了多少其他城市(包括自身)
区间合并
先来讲讲自己初学对区间合并的理解
tree数组记录下区间L,R,并且记录区间内连续前缀1的数量(我记为ll),和连续后缀1的数量(我记为rr)
(Sample1 :1101 那么他的前缀就是2,后缀就是1)
(Sample2 :11111 前缀是5,后缀是5)
向上更新:
给出一个区间[1,7],它是由区间[1,4],[5,7]合成的,
那么[1,7]区间的前缀1到底有多少呢?肯定是包含了[1,4]的前缀1
但是万一[1,4]全是1呢?
那么这时候就要加上[5,7]的前缀1数量了
后缀1同理
void pu(int rt)
{
tree[rt].ll=tree[rt<<1].ll;//肯定包含小区间
tree[rt].rr=tree[rt<<1|1].rr;
if (tree[rt<<1].ll==tree[rt<<1].r-tree[rt<<1].l+1)//特判
tree[rt].ll+=tree[rt<<1|1].ll;
if (tree[rt<<1|1].rr==tree[rt<<1|1].r-tree[rt<<1|1].l+1)
tree[rt].rr+=tree[rt<<1].rr;
}
建树:
建树是没有什么区别的,只是根据题意定义的,在这边把叶子节点全部定义为1(因为长度为1的线段肯定最大也就1,并且初始所有城市都是完好的)
void build(int rt,int l,int r)
{
tree[rt].l=l;
tree[rt].r=r;
if (l==r)
{
tree[rt].ll=tree[rt].rr=1;
return;
}
int mid=(l+r)>>1;
build(lson);
build(rson);
pu(rt);
}
更新:
更新也没有什么难度,要修改的只有一座城市,只要递归找到这座城市并赋值为0,然后向上更新就可以了
void update(int p,int x,int rt,int l,int r)
{
if (tree[rt].l==tree[rt].r)
{
tree[rt].ll=tree[rt].rr=x;
return;
}
int mid=(tree[rt].l+tree[rt].r)>>1;
if (p<=mid)
update(p,x,rt<<1,l,r);
else
update(p,x,rt<<1|1,l,r);
pu(rt);
}
查询
最难的就是查询了,理解了好久
我们一直记录的是一个区间的前缀1和后缀1的数量,那么我们现在题目问的是一个城市连接其他城市的数量,这有什么关系呢?
Sample:110<–1-->101
现在给出7个城市,我们要求第四个城市的连接情况,动态来看就是从这个点往左往右出发,直到碰到0停止,也就是要求一个最长的连续1区间
我们查询的时候一定是查到这个点被包含在连续区间内
一开始查1,7,发现4并不在他的连续区间([1,2],[7,7])内,就继续查左子树(因为4在左子树上)
然后查1,4,发现4在他的连续区间内([1,2],[4,4]),那么此时就可以直接退出了!
找到左子树的后缀和右子树的前缀加起来就可以了
int query(int p,int rt,int l,int r)
{
if (tree[rt].l==tree[rt].r)
{
return tree[rt].ll;
}
int mid=(tree[rt].l+tree[rt].r)>>1;
if (p<=mid)
{
if (tree[rt<<1].r-tree[rt<<1].rr+1<=p)
return tree[rt<<1].rr+tree[rt<<1|1].ll;
query(p,rt<<1,l,r);
}
else
{
if (tree[rt<<1|1].l+tree[rt<<1|1].ll-1>=p)
return tree[rt<<1].rr+tree[rt<<1|1].ll;
query(p,rt<<1|1,l,r);
}
}
完整AC代码
/**
* Author1: low-equipped w_udixixi
* Author2: Sher丶lock
* Date :2019-08-17
**/
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<queue>
#include<map>
#define pb push_back
#define rep(x,a,b) for (int x=a;x<=b;x++)
#define repp(x,a,b) for (int x=a;x<b;x++)
#define pi 3.14159265358979323846
#define mem(a,x) memset(a,x,sizeof a)
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
using namespace std;
const int maxn=2e6+7;
const int INF=1e9;
struct node
{
int l,r;
int ll,rr;
} tree[maxn];
stack<int > S;
void pu(int rt)
{
tree[rt].ll=tree[rt<<1].ll;
tree[rt].rr=tree[rt<<1|1].rr;
if (tree[rt<<1].ll==tree[rt<<1].r-tree[rt<<1].l+1)
{
tree[rt].ll+=tree[rt<<1|1].ll;
}
if (tree[rt<<1|1].rr==tree[rt<<1|1].r-tree[rt<<1|1].l+1)
{
tree[rt].rr+=tree[rt<<1].rr;
}
}
void build(int rt,int l,int r)
{
tree[rt].l=l;
tree[rt].r=r;
if (l==r)
{
tree[rt].ll=tree[rt].rr=1;
return;
}
int mid=(l+r)>>1;
build(lson);
build(rson);
pu(rt);
}
void update(int p,int x,int rt,int l,int r)
{
if (tree[rt].l==tree[rt].r)
{
tree[rt].ll=tree[rt].rr=x;
return;
}
int mid=(tree[rt].l+tree[rt].r)>>1;
if (p<=mid)
{
update(p,x,rt<<1,l,r);
}
else
{
update(p,x,rt<<1|1,l,r);
}
pu(rt);
}
int query(int p,int rt,int l,int r)
{
if (tree[rt].l==tree[rt].r)
{
return tree[rt].ll;
}
int mid=(tree[rt].l+tree[rt].r)>>1;
if (p<=mid)
{
if (tree[rt<<1].r-tree[rt<<1].rr+1<=p)
return tree[rt<<1].rr+tree[rt<<1|1].ll;
query(p,rt<<1,l,r);
}
else
{
if (tree[rt<<1|1].l+tree[rt<<1|1].ll-1>=p)
return tree[rt<<1].rr+tree[rt<<1|1].ll;
query(p,rt<<1|1,l,r);
}
}
int main()
{
int n,m,x;
char s[30];
while(scanf("%d%d",&n,&m)!=EOF)
{
build(1,1,n);
while(!S.empty())
S.pop();
while(m--)
{
scanf("%s",s);
if (s[0]=='D')
{
scanf("%d",&x);
S.push(x);
update(x,0,1,1,n);
}
else if (s[0]=='Q')
{
scanf("%d",&x);
cout<<query(x,1,1,n)<<endl;
}
else if (s[0]=='R')
{
int x=S.top();
S.pop();
update(x,1,1,1,n);
}
}
}
return 0;
}
感觉这只是一个区间合并的小题,稍微写一下增长一下理解吧,欢迎指出错误!