[蓝桥杯2019初赛]后缀表达式

题目描述

给定N 个加号、M 个减号以及N + M + 1 个整数A1,A2,...,AN+M+1
小明想知道在所有由这N 个加号、M 个减号以及N + M +1 个整数凑出的合法的后缀表达式中，结果最大的是哪一个？
请你输出这个最大的结果。
例如使用1 2 3 + -，则“2 3 + 1 -” 这个后缀表达式结果是4，是最大的。

输入

第一行包含两个整数N 和M。
第二行包含N + M + 1 个整数A1,A2,...,AN+M+1
0<=N,M<=100000,-10^9<=Ai<=10^9

输出

输出一个整数，代表答案。

样例输入

1 1

1 2 3

样例输出

4

分析

以输入A, b, c, d为例，大写表示正数，小写表示负数

sum_abs = abs(A) + abs(b) + abs(c) + abs(d);

sum_normal = A + b + c + d;

一、没有减号，即M = 0

       ret = A + b + c + d;

二、减号的数量等于负数的数量，对于输入序列A, b, c, d，即M = 3

       sum = A - b - c - d;

       在每个负数前都加上减号，所以ret = sum_abs;

三、减号的数量小于负数的数量，对于输入序列A, b, c, d，减号M的取值为2，1

    即 sum = A - b - (c + d) 或 sum = A - (b + c) - d 或 sum = A - (b + c + d)

    可以得到一般情况下，ret = sum_abs；

特殊情况：

    如果A改为a，即输入序列全为负数，此时减号M的取值为3，2，1

    sum = a - b - c - d

    或 sum = a - b - (c + d) = a - b - c - d

    或 sum = a - (b + c) - d = a - b - c - d

    或 sum = a - (b + c + d) = a - b - c - d

    a为负数，后面的表达式结果为正数，为使ret最大，a为负数中的最大值

    所以ret = sum_abs + 2 * maxum;

四、减号的数量大于负数的数量，如果输入序列为A，B，C，d，此时减号M的取值为3，2

    即sum = A - (d - B - C) = A + B + C - d 或 sum = A - (d - B) + C = A - d + B + C

    可以得到一般情况下，ret = sum_abs;

特殊情况：

    如果d改为D，即输入序列全为正数，此时减号M的取值为3，2，1

    sum = A - (B - C - D) = A - B + C + D

    或 sum = A - (B - C) + D = A - B + C + D

    或 sum = A - B + C + D

    可以得到结果均为A - B + C + D ，此时B应为非负数的最小值，使ret最大

    所以ret = sum_abs - 2 * minum;

代码实现：

#include <iostream>
#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
int n, m;
int num[200001];
int negative = 0, positive = 0;
int minum = 1e9 + 1, maxum = -1e9 - 1;
int main(){
    cin >> n >> m;
    long long sum_abs = 0, sum = 0, ret = 0;
    for(int i = 0; i < n + m + 1; i++){
        cin >> num[i];
        if(num[i] < 0) negative++;
        else positive++;
        if(num[i] > maxum && num[i] < 0) maxum = num[i];
        if(num[i] < minum && num[i] >= 0) minum = num[i];
    }
    for(int i = 0; i < n + m + 1; i++){
        sum_abs += abs(num[i]);
        sum += num[i];
    }
    if(negative == m){
        ret = sum_abs;
    }else if(m == 0){
        ret = sum;
    }else if(negative < m){
        if(negative == 0){
            ret = sum_abs - 2 * minum;
        }else{
            ret = sum_abs;
        }
    }else{
        if(positive == 0){
            ret = sum_abs + 2 * maxum;
        }else{
            ret = sum_abs;
        }
    }   
    cout << ret << endl;
    return 0;
}