HDOJ 5461.Largest Point（贪心）

2015-9-19

问题简述：

有一个n个数字的整数序列A(t1,t2,,,ti,,tj,,,tn)，要求输出其中满足ti!=tj的a*ti^2+b*tj最大值

原题链接

解题思路：

先把序列A中的数排序，并且再用一个数组保存平方数也排序，这样每次查找最小或最大数的复杂度降为O(1)。之后列举所有的情况（a=0,b=0,a>0&&b>0...)
注意列举的时候要保证ti不等于tj，所以要讨论选择最小最大值是不要重复选择。
我使用的方法是构造两个结构体，分别用来存ti和其平方数，且存在一个pos值记录最初的下标，方便排序后查重。

---

源代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#include <map>
#include <cmath>
#define MAX 5000010
#define ll long long
using namespace std;

int t, n, a, b;

struct node {
    ll val;
    int pos;
} ar[MAX], pr[MAX];

bool cmp( node x, node y ) {
    return x.val < y.val;
}

int main()
{
    scanf( "%d", &t );
    int cas = 1;
    while( t -- ) {
        scanf( "%d%d%d", &n, &a, &b );
        for( int i = 0; i < n; i ++ ) {
            scanf( "%I64d", &ar[i].val );
            ar[i].pos = i;
            pr[i].val = ar[i].val * ar[i].val;
            pr[i].pos = i;
        }
        sort( ar, ar + n, cmp );
        sort( pr, pr + n, cmp );

        ll ans = 0;
        if( a == 0 ) {
            if( b >= 0 )
                ans += b * ar[n-1].val;
            else
                ans += b * ar[0].val;
            printf( "Case #%d: %I64d\n", cas ++, ans );
            continue;
        }
        if( b == 0 ) {
            if( a >= 0 )
                ans += a * pr[n-1].val;
            else
                ans += a * pr[0].val;
            printf( "Case #%d: %I64d\n", cas ++, ans );
            continue;
        }
        if( a > 0 && b > 0 ) {
            ans += a * pr[n-1].val;
            if( pr[n-1].pos == ar[n-1].pos )
                ans += b * ar[n-2].val;
            else
                ans += b * ar[n-1].val;
            ll tmp = b * ar[n-1].val;
            if( pr[n-1].pos == ar[n-1].pos )
                tmp += a * pr[n-2].val;
            else
                tmp += a * pr[n-1].val;
            ans = max( ans, tmp );
        }
        else if( a > 0 && b < 0 ) {
            ans += b * ar[0].val;
            if( ar[0].pos == pr[n-1].pos )
                ans += a * pr[n-2].val;
            else
                ans += a * pr[n-1].val;
            ll tmp = a * pr[n-1].val;
            if( pr[n-1].pos == ar[0].pos )
                tmp += b * ar[1].val;
            else
                tmp += b * ar[0].val;
            ans = max( tmp, ans );
        }
        else if( a < 0 && b > 0 ) {
            ans += b * ar[n-1].val;
            if( ar[n-1].pos == pr[0].pos )
                ans += a * pr[1].val;
            else
                ans += a * pr[0].val;
            ll tmp = a * pr[0].val;
            if( pr[0].pos == ar[n-1].pos )
                tmp += b * ar[n-2].val;
            else
                tmp += b * ar[n-1].val;
            ans = max( tmp, ans );
        }
        else {
            ans += b * ar[0].val;
            if( ar[0].pos == pr[0].pos )
                ans += a * pr[1].val;
            else
                ans += a * pr[0].val;
            ll tmp = a * pr[0].val;
            if( pr[0].pos == ar[0].pos )
                tmp += b * ar[1].val;
            else
                tmp += b * ar[0].val;
            ans = max( tmp, ans );
        }


        printf( "Case #%d: %I64d\n", cas ++, ans );
    }
    return 0;
}