hdu4831 Scenic Popularity（线段树）

　　题目地址：http://acm.hdu.edu.cn/showproblem.php?pid=4831

　　题目大概意思就是有多个风景区和休息区，每个风景区有热度，休息区的热度与最接近的分景区的热度相同，题目要求求出小于等于给定热度值的风景区和休息区的个数。显然如果直接暴力的话，复杂度为O(TKN)，达到10^9次方数量级，复杂度过高，对于这种问答的题目，最一般的思路其实是线段树，线段树更改和查询的时间复杂度均为O(logn)，所以如果用线段树的话，这道题目的复杂度为O(TKlogH)，达到10^5左右的数量级，这个复杂度可以满足题目要求。

　　线段树的实现一般用完全二叉树来实现，每个节点都可以看做是一段"线段"或者某个”范围“(x,y]，这道题目节点表示的”范围“就是某段景区热度值(x,y]，同时节点还记录了在该”范围“中包含的景区和休息区的个数，所以树的叶子节点就是等于某个热度值的景区和休息区的个数。线段树的定义如下:

#define MAX_SIZE 100000
typedef struct _tree
{
    int count;
    int left, right;
}TreeNode;
TreeNode tree[ * MAX_SIZE + ];
void Init(int x, int left, int right)
{
    tree[x].count = ;
    tree[x].left = left;
    tree[x].right = right;
    if (left +  == right)
        return;
    Init(x << , left, (left + right) >> );
    Init((x << ) + , (left + right) >> , right);
}
void Update(int x, int coord, int count)
{
    int left = tree[x].left;
    int right = tree[x].right;
    int mid = (left + right) >> ;
    if (left +  == right)
    {
        tree[x].count += count;
        return;
    }
    if (coord <= mid)
        Update(x << , coord, count);
    else
        Update((x << ) + , coord, count);
    tree[x].count += count;
}
int Query(int x, int coord)
{
    int left = tree[x].left;
    int right = tree[x].right;
    int mid = (left + right) >> ;
    if (left +  == right)
        return tree[x].count;
    if (coord <= mid)
        return Query(x << , coord);
    else if (coord > mid)
        return tree[x << ].count + Query((x << ) + , coord);
}

　　每次更新某个景区的热度值时，受影响的是该景区和离该景区近的休息区的热度值，其它的景区和热度值是保持不变的，只要预处理离每个景区近的休息区的个数，就可以在更新某个景区的热度值后立马得到哪些景区和休息区的热度值受到影响，从而更新线段树。不过需要注意的是某个休息区同时与两个景区距离相同的情况。我的方法是对每个景区和休息区设置两个变量li和ri，如果是景区，li表示向前所能影响的最前的休息区，ri表示向后所能影响的最后的休息区；如果是休息区，并且有两个风景区到该休息区的距离相等，li和ri分别表示这两个风景区。有个这两个变量，对于给定的风景区，就能直接计算该风景区所影响的休息区个数。

typedef struct _xh
{
    int x;
    int h;
    int li, ri;
}XH;
XH data[];

int GetCount(int x)
{
    int cnt = data[x].ri - data[x].li + ;
    if (data[x].li < x && data[data[x].li].li != - && data[data[data[x].li].li].h >= data[x].h)
        cnt--;
    if (data[x].ri > x && data[data[x].ri].ri != - && data[data[data[x].ri].ri].h > data[x].h)
        cnt--;
    return cnt;
}

　　有了上面的分析，对于给定的热度值获取小于等于这个热度值的景区和休息区的个数就简单了，直接查询线段树就可以了。

　　具体代码如下：

 #include <stdio.h>
 #include <string.h>

 #define MAX_SIZE 100000
 typedef struct _tree
 {
     int count;
     int left, right;
 }TreeNode;
 TreeNode tree[ * MAX_SIZE + ];
 typedef struct _xh
 {
     int x;
     int h;
     int li, ri;
 }XH;
 XH data[];

 typedef struct _qy
 {
     char op;
     int x, h;
 }QY;
 QY qy[];

 void Init(int x, int left, int right)
 {
     tree[x].count = ;
     tree[x].left = left;
     tree[x].right = right;
     if (left +  == right)
         return;
     Init(x << , left, (left + right) >> );
     Init((x << ) + , (left + right) >> , right);
 }
 void Update(int x, int coord, int count)
 {
     int left = tree[x].left;
     int right = tree[x].right;
     int mid = (left + right) >> ;
     if (left +  == right)
     {
         tree[x].count += count;
         return;
     }
     if (coord <= mid)
         Update(x << , coord, count);
     else
         Update((x << ) + , coord, count);
     tree[x].count += count;
 }
 int Query(int x, int coord)
 {
     int left = tree[x].left;
     int right = tree[x].right;
     int mid = (left + right) >> ;
     if (left +  == right)
         return tree[x].count;
     if (coord <= mid)
         return Query(x << , coord);
     else if (coord > mid)
         return tree[x << ].count + Query((x << ) + , coord);
 }
 int GetCount(int x)
 {
     int cnt = data[x].ri - data[x].li + ;
     if (data[x].li < x && data[data[x].li].li != - && data[data[data[x].li].li].h >= data[x].h)
         cnt--;
     if (data[x].ri > x && data[data[x].ri].ri != - && data[data[data[x].ri].ri].h > data[x].h)
         cnt--;
     return cnt;
 }
 int main(void)
 {
     int t, n, k, i, j, pl, pr, cnt, max, x, h, cntpre, cntnext, cntnew, cntnpre, cntnnext;
     char op;

     scanf("%d", &t);
     for (i = ; i < t; i++)
     {
         max = ;
         memset(data, -, sizeof(data));
         scanf("%d", &n);
         for (j = ; j < n; j++)
         {
             scanf("%d%d", &data[j].x, &data[j].h);
             if (max < data[j].h)
                 max = data[j].h;
         }
         scanf("%d", &k);
         for (j = ; j < k; j++)
         {
             scanf("%1s", &qy[j].op);
             if (qy[j].op == 'Q')
                 scanf("%d", &qy[j].h);
             else
                 scanf("%d%d", &qy[j].x, &qy[j].h);
             if (max < qy[j].h)
                 max = qy[j].h;
         }
         pl = pr = ;
         while (pl < n && !data[pl].h)
             pl++;
         data[pl].li = ;
         pr = pl + ;
         while (pl < n)
         {
             while (pr < n && !data[pr].h)
                 pr++;
             if (pr == n)
                 break;
             if (pl +  == pr)
             {
                 data[pl].ri = pl;
                 data[pr].li = pr;
             }
             else
             {
                 j = pl + ;
                 while (j < pr)
                     if (data[j]. x - data[pl].x < data[pr].x - data[j].x)
                     {
                         data[pl].ri = j;
                         data[pr].li = j + ;
                         j++;
                     }
                     else if (data[j]. x - data[pl].x == data[pr].x - data[j].x)
                     {
                         data[pl].ri = data[pr].li = j;
                         data[j].li = pl;
                         data[j].ri = pr;
                         break;
                     }
                     else
                     {
                         data[pr].li = j;
                         data[pl].ri = j - ;
                         break;
                     }
             }
             pl = pr++;
         }
         if (pl < n)
             data[pl].ri = n - ;
         Init(, , max);
         for (j = ; j < n; j++)
             if (data[j].h)
             {
                 cnt = GetCount(j);
                 Update(, data[j].h, cnt);
             }
         printf("Case #%d:\n", i + );
         for (j = ; j < k; j++)
             if (qy[j].op == 'Q')
                 printf("%d\n", Query(, qy[j].h));
             else
             {
                 x = qy[j].x;
                 h = qy[j].h;
                 cnt = GetCount(x);
                 Update(, data[x].h, -cnt);
                 cntpre = cntnext = cntnpre = cntnnext = ;
                 if (data[x].li < x && data[data[x].li].li != -)
                     cntpre = GetCount(data[data[x].li].li);
                 if (data[x].ri > x && data[data[x].ri].ri != -)
                     cntnext = GetCount(data[data[x].ri].ri);
                 data[x].h = h;
                 cntnew = GetCount(x);
                 Update(, data[x].h, cntnew);
                 if (data[x].li < x && data[data[x].li].li != -)
                     cntnpre = GetCount(data[data[x].li].li);
                 if (data[x].ri > x && data[data[x].ri].ri != -)
                     cntnnext = GetCount(data[data[x].ri].ri);
                 if (cntpre != cntnpre)
                     Update(, data[data[data[x].li].li].h, cntnpre - cntpre);
                 if (cntnext != cntnnext)
                     Update(, data[data[data[x].ri].ri].h, cntnnext - cntnext);
             }
     }
     return ;
 }