HDU 5461 Largest Point (水)

网络赛水题,主要是对t[i]平方,和t[i]分别排序,,然后先根据b的正负判断b该与哪个数相乘,然后在给a找数乘。

#pragma warning(disable:4996)
#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
const int N = 6000005;
typedef long long LL;
LL t[N];
int n;
LL a, b;
vector<LL>vec;

int main(){
    int T; scanf("%d", &T);
    int kase = 1;
    while (T--){
        scanf("%d %lld %lld", &n, &a, &b);
        vec.clear();
        for (int i = 1; i <= n; i++){
            int x;
            scanf("%d", &x);
            t[i] = (LL)x;
            vec.push_back(t[i] * t[i]);
        }
        sort(t + 1, t + 1 + n);
        sort(vec.begin(), vec.end());
        LL ans = 0;

        if (a > 0 && b > 0){
            if (vec[vec.size() - 1] == t[n] * t[n]) {
                ans = vec[vec.size() - 2] * a + b*t[n];
            }
            else{
                ans = vec[vec.size() - 1] * a + b*t[n];
            }
            ans = max(ans, a*t[n] * t[n] + b*t[n - 1]);
        }
        else if (a > 0 && b < 0) {
            if (vec[vec.size() - 1] == t[1] * t[1]) {
                ans = vec[vec.size() - 2] * a + b*t[1];
            }
            else{
                ans = vec[vec.size() - 1] * a + b*t[1];
            }
            ans = max(ans, a*t[1] * t[1] + b*t[2]);
        }
        else if (a<0 && b>0) {
            if (vec[0] == t[n] * t[n]){
                ans = vec[1] * a + b*t[n];
            }
            else{
                ans = vec[0] * a + b*t[n];
            }
            ans = max(ans, a*t[n] * t[n] + b*t[n - 1]);
        }
        else if (a < 0 && b < 0){
            if (vec[0] == t[1] * t[1]){
                ans = vec[1] * a + b*t[1];
            }
            else{
                ans = vec[0] * a + b*t[1];
            }
            ans = max(ans, a*t[1] * t[1] + b*t[2]);
        }
        else if (a == 0) {
            if (b >= 0)ans = b*t[n];
            else ans = b*t[1];
        }
        else{
            if (a > 0)ans = vec[vec.size() - 1] * a;
            else ans = vec[0] * a;
        }
        printf("Case #%d: %lld\n", kase++, ans);
    }
    return 0;
}


你可能感兴趣的:(HDU 5461 Largest Point (水))