题目描述
给定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,-109<=Ai<=109
输出
输出一个整数,代表答案。
样例输入 Copy
1 1
1 2 3
样例输出 Copy
4
这题坑真多。
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
ll a[200010];
int main() {
int n, m, sum;
cin >> n >> m;
sum = n + m + 1;
int neg = 0;//负数个数
for(int i =1 ; i <= sum; i++) {
cin >> a[i];
if(a[i] < 0) neg++;
}
sort(a + 1, a + 1 + sum);
ll ans = 0;
if(m == 0) {//若没有负号
for(int i = 1; i <= sum; i++) ans += a[i];
}
else {
if(neg == 0) {//若没有负数
ans -= a[1];
for(int i = 2; i <= sum; i++) ans += abs(a[i]);
}
else {
if(neg == sum) {//若全为负数
for(int i = 1; i < sum; i++) ans += abs(a[i]);
ans += a[sum];
}
else {
for(int i = 1; i <= sum; i++) ans += abs(a[i]);
}
}
}
cout << ans;
return 0;
}