1. 程式人生 > >HDU-5461 Largest Point 貪心

HDU-5461 Largest Point 貪心

題目連結

題意:

給你n,a,b,n代表有n個t,然後輸入n個t;

讓你求出  a*ti*ti+b*tj(i不等於j) 的最大值;

本來以為是分類討論的問題,呼哧呼哧的寫了幾十行程式碼WA掉了,心有不甘去查了查題解,發現這題根本就不需要分類討論呀。對每個t分別求出a*t*t的值存在aa[]陣列中,b*t的值存在bb[]陣列中。然後由大到小排序,如果aa[]最大值和bb[]最大值對應的t不是同一個t,直接輸出答案,否則就是aa的次大值加上bb的最大值  與 aa的最大值加上bb的次大值的比較。

AC程式碼:

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
using namespace std;
const int MAX=5e6+7;
typedef long long ll;
ll t,n,a,b,arr;
struct node
{
    ll id;
    ll data;
}aa[MAX],bb[MAX];
bool cmp(node a,node b)
{
    return a.data>b.data;
}
int main()
{
    scanf("%lld",&t);
    for(ll o=1; o<=t; o++)
    {
        ll ans=0;
        scanf("%lld%lld%lld",&n,&a,&b);
        for(ll i=0; i<n; i++)
        {
            scanf("%lld",&arr);
            aa[i].id=arr;
            aa[i].data=a*arr*arr;
            bb[i].id=arr;
            bb[i].data=b*arr;
        }
        sort(aa,aa+n,cmp);
        sort(bb,bb+n,cmp);
        if(aa[0].id!=bb[0].id)
            ans=aa[0].data+bb[0].data;
        else
            ans=max(aa[0].data+bb[1].data,aa[1].data+bb[0].data);
        printf("Case #%lld: %lld\n",o,ans);
    }
    return 0;
}