蓝桥杯-算法训练--ALGO-8 操作格子

问题描述

有n个格子，从左到右放成一排，编号为1-n。

共有m次操作，有3种操作类型：

1.修改一个格子的权值，

2.求连续一段格子权值和，

3.求连续一段格子的最大值。

对于每个2、3操作输出你所求出的结果。

输入格式

第一行2个整数n，m。

接下来一行n个整数表示n个格子的初始权值。

接下来m行，每行3个整数p,x,y，p表示操作类型，p=1时表示修改格子x的权值为y，p=2时表示求区间[x,y]内格子权值和，p=3时表示求区间[x,y]内格子最大的权值。
输出格式

有若干行，行数等于p=2或3的操作总数。

每行1个整数，对应了每个p=2或3操作的结果。

样例输入
4 3
1 2 3 4
2 1 3
1 4 3
3 1 4
样例输出
6
3
数据规模与约定

对于20%的数据n <= 100，m <= 200。

对于50%的数据n <= 5000，m <= 5000。

对于100%的数据1 <= n <= 100000，m <= 100000，0 <= 格子权值 <= 10000

用线段树来解题，一开始的时候结点空间开小了= =

因为N个格子，用线段树的话一共会有2*N-1个结点，所以我不小心就开了2*N-1个结点的空间

结果。。。一半超时。。

修改后发现还是有一个用例超时，上代码：

#include<stdio.h>
#include<iostream>
using namespace std;
const int MAX_N = 100005;
#define max(a,b) a>b?a:b
struct NODE{
    int left;   //左子树
    int right;  //右子树
    int totalValue;  //总和
    int maxValue;     //最大值
}node[10 * MAX_N];
int nodeValue[MAX_N];
//建树
void buildTree(int i, int left, int right){
    node[i].left = left;
    node[i].right = right;
    if (left == right){
        node[i].maxValue = nodeValue[left];
        node[i].totalValue = nodeValue[left];
    }
    else{
        buildTree(2 * i, left, (left + right) / 2);
        buildTree(2 * i + 1, (left + right) / 2 + 1, right);
        node[i].maxValue = node[2 * i].maxValue > node[2 * i + 1].maxValue ? node[2 * i].maxValue : node[2 * i + 1].maxValue;
        node[i].totalValue = node[2 * i].totalValue + node[2 * i + 1].totalValue;
    }
}

//区间更新
void upDate(int i, int x, int changedX){
    if (node[i].left == node[i].right){
        node[i].maxValue = changedX;
        node[i].totalValue = changedX;
    }
    else{
        if (x <= (node[i].left + node[i].right) / 2)
            upDate(2 * i, x, changedX);
        else if (x >= (node[i].left + node[i].right) / 2)
            upDate(2 * i + 1, x, changedX);
        node[i].maxValue = node[2 * i].maxValue > node[2 * i + 1].maxValue ? node[2 * i].maxValue : node[2 * i + 1].maxValue;
        node[i].totalValue = node[2 * i].totalValue + node[2 * i + 1].totalValue;
    }
}

//查找区间最大值
//i表示node[i]结点，left,right表示查找范围
int findMax(int i, int left, int right){

    int maxValue = -1;
    if (node[i].left == left && node[i].right == right){        //完全重合
        maxValue = max(maxValue, node[i].maxValue);
        return maxValue;
    }
    if (left <= node[2 * i].right){         //范围跟node[i]的左子树有交集
        if (right <= node[2 * i].right){
            maxValue = max(maxValue, findMax(2 * i, left, right));
        }
        else{
            maxValue = max(maxValue, findMax(2 * i, left, node[2 * i].right));
        }
    }
    if (right >= node[2 * i + 1].left){          //范围跟node[i]的右子树有交集
        if (left >= node[2 * i + 1].left){     //被右子树完全包含
            maxValue = max(maxValue, findMax(2 * i + 1, left, right));
        }
        else{
            maxValue = max(maxValue, findMax(2 * i + 1, node[2 * i + 1].left, right));
        }
    }
    return maxValue;
}

//查找区间数值之和
int findTotal(int i, int left, int right){
    int total = 0;
    if (node[i].left == left && node[i].right == right){
        total = node[i].totalValue;
        return total;
    }

    if (left <= node[2 * i].right){
        if (right <= node[2 * i].right){
            total = findTotal(2 * i, left, right);
        }
        else{
            total += findTotal(2 * i, left, node[2 * i].right);
        }
    }
    if (right >= node[2 * i + 1].left){
        if (left >= node[2 * i + 1].left){
            total = findTotal(2 * i + 1, left, right);
        }
        else{
            total += findTotal(2 * i + 1, node[2 * i + 1].left, right);
        }
    }

    return total;
}
int main(){
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++)
        scanf("%d", &nodeValue[i]);
    buildTree(1, 1, n);

    for (int j = 1; j <= m; j++){
        int workIndex, x, y;
        scanf("%d%d%d", &workIndex, &x, &y);
        switch (workIndex){
        case 1:
            upDate(1, x, y);
            break;
        case 2:
            printf("%d\n", findTotal(1, x, y));
            break;
        case 3:
            printf("%d\n", findMax(1, x, y));
            break;
        }
    }

然后我修改了一下，不要在每一个查找区间数值方法里面比较最大值和总和，而是在left == right的时候才比较最大值和总和。

AC代码：

#include<stdio.h>
#include<iostream>
using namespace std;
const int MAX_N = 100005;
#define max(a,b) a>b?a:b
struct NODE{
    int left;   //左子树
    int right;  //右子树
    int totalValue;  //总和
    int maxValue;     //最大值
}node[10 * MAX_N];
int nodeValue[MAX_N];

int maxValue = -1;
int totalValue = 0;
//建树
void buildTree(int i, int left, int right){
    node[i].left = left;
    node[i].right = right;
    if (left == right){
        node[i].maxValue = nodeValue[left];
        node[i].totalValue = nodeValue[left];
    }
    else{
        buildTree(2 * i, left, (left + right) / 2);
        buildTree(2 * i + 1, (left + right) / 2 + 1, right);
        node[i].maxValue = node[2 * i].maxValue > node[2 * i + 1].maxValue ? node[2 * i].maxValue : node[2 * i + 1].maxValue;
        node[i].totalValue = node[2 * i].totalValue + node[2 * i + 1].totalValue;
    }
}

//区间更新
void upDate(int i, int x, int changedX){
    if (node[i].left == node[i].right){
        node[i].maxValue = changedX;
        node[i].totalValue = changedX;
    }
    else{
        if (x <= (node[i].left + node[i].right) / 2)
            upDate(2 * i, x, changedX);
        else if (x >= (node[i].left + node[i].right) / 2)
            upDate(2 * i + 1, x, changedX);
        node[i].maxValue = node[2 * i].maxValue > node[2 * i + 1].maxValue ? node[2 * i].maxValue : node[2 * i + 1].maxValue;
        node[i].totalValue = node[2 * i].totalValue + node[2 * i + 1].totalValue;
    }
}

//查找区间最大值
//i表示node[i]结点，left,right表示查找范围
void findMax(int i, int left, int right){

    if (node[i].left == left && node[i].right == right){        //完全重合
        maxValue = max(maxValue, node[i].maxValue);
        return;
    }
    if (left <= node[2 * i].right){         //范围跟node[i]的左子树有交集
        if (right <= node[2 * i].right){
            findMax(2 * i, left, right);
        }
        else{
            findMax(2 * i, left, node[2 * i].right);
        }
    }
    if (right >= node[2 * i + 1].left){          //范围跟node[i]的右子树有交集
        if (left >= node[2 * i + 1].left){     //被右子树完全包含
            findMax(2 * i + 1, left, right);
        }
        else{
            maxValue, findMax(2 * i + 1, node[2 * i + 1].left, right);
        }
    }
}

//查找区间数值之和
void findTotal(int i, int left, int right){
    if (node[i].left == left && node[i].right == right){
        totalValue += node[i].totalValue;
        return;
    }

    if (left <= node[2 * i].right){
        if (right <= node[2 * i].right){
            findTotal(2 * i, left, right);
        }
        else{
            findTotal(2 * i, left, node[2 * i].right);
        }
    }
    if (right >= node[2 * i + 1].left){
        if (left >= node[2 * i + 1].left){
            findTotal(2 * i + 1, left, right);
        }
        else{
            findTotal(2 * i + 1, node[2 * i + 1].left, right);
        }
    }
}
int main(){
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++)
        scanf("%d", &nodeValue[i]);
    buildTree(1, 1, n);

    for (int j = 1; j <= m; j++){
        int workIndex, x, y;
        scanf("%d%d%d", &workIndex, &x, &y);
        switch (workIndex){
        case 1:
            upDate(1, x, y);
            break;
        case 2:
            findTotal(1, x, y);
            printf("%d\n", totalValue);
            totalValue = 0;
            break;
        case 3:
            findMax(1, x, y);
            printf("%d\n", maxValue);
            maxValue = -1;
            break;
        }
    }

    /*
    for(int j = 1;j<=2*n-1;j++){
    cout<<"left: "<<node[j].left<<endl;
    cout<<"right: "<<node[j].right<<endl;
    cout<<"maxValue: "<<node[j].maxValue<<endl;
    cout<<"totalValue: "<<node[j].totalValue<<endl;
    }
    */
    /*
    cout<<"1 到 4号格子最大值： "<<findMax(1,1,4)<<endl;
    cout<<"1 到 2号格子最大值： "<<findMax(1,1,2)<<endl;
    cout<<"1 到 3号格子最大值： "<<findMax(1,1,3)<<endl;
    cout<<"2 到 4号格子最大值： "<<findMax(1,2,4)<<endl;
    cout<<"2 到 3号格子最大值： "<<findMax(1,2,3)<<endl;
    cout<<"3 到 4号格子最大值： "<<findMax(1,3,4)<<endl;

    cout<<"1 到 4号格子权值和： "<<findTotal(1,1,4)<<endl;
    cout<<"1 到 2号格子权值和： "<<findTotal(1,1,2)<<endl;
    cout<<"1 到 3号格子权值和： "<<findTotal(1,1,3)<<endl;
    cout<<"2 到 4号格子权值和： "<<findTotal(1,2,4)<<endl;
    cout<<"2 到 3号格子权值和： "<<findTotal(1,2,3)<<endl;
    cout<<"3 到 4号格子权值和： "<<findTotal(1,3,4)<<endl;
    */
}