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;
}