hdoj1754-I Hate It && 1166-敵兵佈陣(線段樹)

題目鏈接

Problem Description

很多學校流行一種比較的習慣。老師們很喜歡詢問，從某某到某某當中，分數最高的是多少。
這讓很多學生很反感。

不管你喜不喜歡，現在需要你做的是，就是按照老師的要求，寫一個程序，模擬老師的詢問。當然，老師有時候需要更新某位同學的成績。

Input

本題目包含多組測試，請處理到文件結束。
在每個測試的第一行，有兩個正整數 N 和 M ( 0< N<=200000,0< M<5000 )，分別代表學生的數目和操作的數目。
學生ID編號分別從1編到N。
第二行包含N個整數，代表這N個學生的初始成績，其中第i個數代表ID爲i的學生的成績。
接下來有M行。每一行有一個字符 C (只取’Q’或’U’) ，和兩個正整數A，B。
當C爲’Q’的時候，表示這是一條詢問操作，它詢問ID從A到B(包括A,B)的學生當中，成績最高的是多少。
當C爲’U’的時候，表示這是一條更新操作，要求把ID爲A的學生的成績更改爲B。

Output

對於每一次詢問操作，在一行裏面輸出最高成績。

Sample Input

5 6
1 2 3 4 5
Q 1 5
U 3 6
Q 3 4
Q 4 5
U 2 9
Q 1 5

Sample Output

5
6
5
9

思路

      此題屬於線段樹求區間最大值的模板題，詳解見大神MetalSeed博客;

code

#include <iostream>
#include <fstream>
using namespace std;

const int MAX = 200000+5;

class Node      //節點
{
        public:
                int l, r, v; //區間的左右邊界及區間內的最大值V
};

Node node[4*MAX];
int grade[MAX];
int n, m;

void creat(int root, int l, int r)
{
        node[root].l = l;
        node[root].r = r;
        if(l == r)
        {
                node[root].v = grade[l];
                return ;
        }
        int m = l + (r - l) / 2;
        creat(root * 2, l, m);
        creat(root * 2 + 1, m + 1, r);
        node[root].v = node[root*2].v > node[root*2+1].v? node[root*2].v : node[root*2+1].v;
}

void update(int root, int x, int _v)
{
        if(node[root].l == node[root].r)
        {
                node[root].v = _v;
                return ;
        }
        if(x <= node[root*2].r)
        {
                update(root*2, x, _v);
        }
        else
        {
                update(root*2+1, x, _v);
        }
        node[root].v = node[root*2].v > node[root*2+1].v? node[root*2].v : node[root*2+1].v;
}

int query(int root, int l, int r)
{
        if(node[root].l == l && node[root].r == r)
        {
                return node[root].v;
        }
        int s;
        if(r <= node[root*2].r)
        {
                s = query(root*2, l, r);
        }
        else if(l >= node[root*2+1].l)
        {
                s = query(root*2+1, l, r);
        }
        else
        {
                int s1 = query(root*2, l, node[root*2].r);
                int s2 = query(root*2+1, node[root*2+1].l, r);
                s = s1 > s2? s1 : s2;
        }
        return s;
}

int main()
{
        //ifstream cin("data.in");
        std::ios::sync_with_stdio(false);
        std::cin.tie(0);
        while(cin >> n >> m)
        {
                for(int i = 1; i <= n; ++ i)
                {
                        cin >> grade[i];
                }
                creat(1, 1, n);
                for(int i = 0; i < m; ++ i)
                {
                        char ch;
                        cin >> ch;
                        if(ch == 'U')
                        {
                                int x, v;
                                cin >> x >> v;
                                update(1, x, v);
                        }
                        else
                        {
                                int l, r;
                                cin >> l >> r;
                                cout << query(1, l, r) << endl;
                        }
                }
        }
    return 0;
}

hdoj1166-敵兵佈陣

      此題屬於球區間的數值和問題；

code：

#include <iostream>
#include <algorithm>
#include <string>
#include <fstream>
#include <cstdio>

using namespace std;

class Node
{
        public:
                int l, r, v;
};
const int MAX = 50000+5;

Node segTree[4*MAX];
int data[MAX];

void creat(int root, int l, int r)
{
        segTree[root].l = l;
        segTree[root].r = r;
        if(l == r)
        {
                segTree[root].v = data[l];

        }
        else
        {
                int mid = (l + r) >> 1;
                creat(2*root, l, mid);
                creat(2*root+1, mid+1, r);
                segTree[root].v = segTree[2*root].v + segTree[2*root+1].v;
        }
}

void update(int root, int x, int _v)
{
        //cout << _v << endl;
        int l = segTree[root].l;
        int r = segTree[root].r;
        if(l == r)
        {
                segTree[root].v += _v;
        }
        else
        {
                int mid = (l + r) >> 1;
                if(x <= mid)
                {
                        update(2*root, x, _v);
                }
                else
                {
                        update(2*root+1, x, _v);
                }
                segTree[root].v = segTree[2*root].v + segTree[2*root+1].v;
        }
}

int query(int root, int l, int r)
{
        if(segTree[root].l >= l && segTree[root].r <= r)
        {
                return segTree[root].v;
        }
        else
        {
                int q1 = 0, q2 = 0;
                //cout << segTree[2*root].r << endl;
                 if(r <= segTree[2*root].r)
                {
                        q1 = query(2*root, l, r);
                }
                else if(l >= segTree[2*root+1].l)
                {
                        q2 = query(2*root+1, l, r);
                }
                else
                {
                        q1 = query(2*root, l, r);
                        q2 = query(2*root+1, l, r);
                }
                return q1 + q2;
        }
}

int main()
{
        //freopen("data.in", "r", stdin);
        int t, cnt = 1;
        scanf("%d", &t);
        while(t -- )
        {
                printf("Case %d:\n", cnt ++);
                int n;
                scanf("%d", &n);
                for(int i = 1; i <= n; ++ i)
                {
                        scanf("%d", &data[i]);
                }
                creat(1, 1, n);

                char ch[10];
                while(~scanf("%s", ch) && ch[0] != 'E')
                {
                        if(ch[0] == 'Q')
                        {
                                int l, r;
                                scanf("%d%d", &l, &r);
                                printf("%d\n", query(1, l, r));
                        }
                        else if(ch[0] == 'A')
                        {
                                int x, v;
                                scanf("%d%d", &x, &v);
                                update(1, x, v);
                        }
                        else if(ch[0] == 'S')
                        {
                                int x, v;
                                scanf("%d%d", &x, &v);
                                update(1, x, -v);
                        }
                }
        }
        return 0;
}