线段树基础之区间合并HDU1540

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;
}

感觉这只是一个区间合并的小题，稍微写一下增长一下理解吧，欢迎指出错误！