【蓝桥杯第十届真题】试题 I: 后缀表达式 Java版

试题 I: 后缀表达式

时间限制: 1.0s 内存限制: 512.0MB 本题总分：25 分

【问题描述】
给定 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。
【输出格式】
输出一个整数，代表答案。
【样例输入】
1 1
1 2 3
【样例输出】
4
【评测用例规模与约定】
对于所有评测用例，0 ≤ N, M ≤ 100000，−109 ≤ Ai ≤ 109。

分析
将所有数降序排序，然后把排在前面的数值大的数相加，再减去后面数值较小的数。

Java 代码：

import java.io.*;
import java.util.Arrays;
import java.util.Collections;

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));

        String[] s=in.readLine().split(" ");
        int n=Integer.parseInt(s[0]);
        int m=Integer.parseInt(s[1]);

        String[] ss=in.readLine().split(" ");
        Integer[] a=new Integer[n+m+1];
        for (int i=0;i<n+m+1;i++){
            a[i]=Integer.parseInt(ss[i]);
        }
        Arrays.sort(a, Collections.reverseOrder());

        int sum=a[0];
        int num=n+m+1;
        for (int i=1;i<num;i++){
            if (n>0){
                sum+=a[i];
                n--;
            }else if (m>0){
                sum-=a[i];
                m--;
            }
        }
        System.out.println(sum);
    }
}