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

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