HDU 1540 Tunnel Warfare

线段树单点更新 区间合并

题目描述：一串线性的村庄，除了头尾的两个村庄，其他村庄都和左右的两个村庄相邻，可以摧毁一个村庄，这样连接的关系就会被破坏，也可以修复一个村庄，连接的关系也会修复，询问一个村庄，求与它相连接的村庄有多少个。

解题分析：借鉴了kuangbin的想法，维护一个节点的最大左连续区间，最大右连续区间和最大的连续区间。当询问的时候，寻找k所在的那个最大连续区间。

代码如下：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <stack>
using namespace std;

const int maxn = 50000 + 10;
struct Node
{
    int l,r;//结点的左边界和右边界
    int ls,rs;//以左边界为起点的最大连续区间，以右边界为终点的最大连续区间.
    int sum;//节点内的最大连续区间
} P[maxn<<2];
int n,m;

void pushup(int pos)//向上更新父亲节点
{
    P[pos].ls = P[pos<< 1].ls;//更新结点的左区间
    P[pos].rs = P[pos<< 1 | 1].rs;//更新结点的右区间
    if(P[pos << 1].ls == (P[pos<< 1].r - P[pos<< 1].l + 1))
    {//如果左区间就是左子节点的区间长度，再加上右子节点的左区间
        P[pos].ls += P[pos<<1 | 1].ls;
    }
    if(P[pos << 1 | 1].rs == ( P[pos << 1 | 1].r - P[pos << 1 | 1].l + 1))
    {//如果右区间就是右子节点的区间长度，再加上左子节点的右区间
        P[pos].rs += P[pos<<1].rs;
    }
    P[pos].sum = max(max(P[pos].ls,P[pos].rs),P[pos<< 1].rs + P[pos << 1 | 1].ls);//更新结点最大连续区间
}

void build(int l,int r,int pos)//建树
{
    P[pos].ls = P[pos].rs = P[pos].sum = r - l + 1;
    P[pos].l = l;
    P[pos].r = r;
    if(l == r)
    {
        return;
    }
    int mid = (l + r) / 2;
    build(l,mid,pos<<1);
    build(mid + 1,r,pos<< 1 | 1);
}

void update(int pos,int k,int v)//当前结点的编号，要修改的结点编号，修改的值
{
    if(P[pos].l == P[pos].r)
    {
        P[pos].ls = P[pos].rs = P[pos].sum = v;
        return;
    }
    int mid = (P[pos].l + P[pos].r) / 2;
    if(k <= mid) update(pos << 1,k,v);
    else if(k > mid) update(pos << 1 | 1,k,v);
    pushup(pos);
}

int query(int pos,int k)//当前结点的编号，查询结点的编号
{//当为叶子结点或区间都不连续或整个区间连续
    if(P[pos].l == P[pos].r || P[pos].sum == 0 || P[pos].sum == (P[pos].r - P[pos].l + 1))
        return P[pos].sum;
    int ans = 0;
    int mid = (P[pos].l + P[pos].r) / 2;
    if(k <= mid)
    {
        if(k >= (P[pos << 1].r - P[pos << 1].rs + 1))//当k在左结点的右区间内
        {
            ans = P[pos << 1 | 1].ls + P[pos << 1].rs;//答案是左结点的右区间加上右结点的左区间
        }
        else
            ans =  query(pos << 1,k);
    }
    else
    {
        if(k <= (P[pos << 1 | 1].l + P[pos << 1 | 1].ls - 1))//当k在右结点的左区间内
        {
            ans = P[pos << 1].rs + P[pos << 1 | 1].ls;//答案是左结点的右区间加上右结点的左区间
        }
        else
            ans = query(pos << 1 | 1,k);
    }
    return ans;
}

int main()
{
    while(scanf("%d %d",&n,&m) != EOF)
    {
        stack<int> last;
        build(1,n,1);
        while(m--)
        {
            char c;
            int k;
            cin >> c;
            if(c == 'D')
            {
                scanf("%d",&k);
                last.push(k);
                update(1,k,0);
            }
            else if(c == 'Q')
            {
                scanf("%d",&k);
                cout << query(1,k) << endl;
            }
            else if(c == 'R')
            {
                if(!last.empty())
                {
                    k = last.top();
                    last.pop();
                    update(1,k,1);
                }
            }
        }
    }
    return 0;
}

第一次做到区间和并的线段树题目，学习了。