CodeForces 1406B Maximum Product

CodeForces 1406B Maximum Product

周赛的签到题。

题目大意
每组数据给出5个或以上的整数，求从中选出5个数，使它们的乘积最大。

样例输入
4
5
-1 -2 -3 -4 -5
6
-1 -2 -3 1 2 -1
6
-1 0 0 0 -1 -1
6
-9 -7 -5 -3 -2 1

样例输出
-120
12
0
945

样例解释
In the first test case, choosing a1,a2,a3,a4,a5 is a best choice: (−1)⋅(−2)⋅(−3)⋅(−4)⋅(−5)=−120.

In the second test case, choosing a1,a2,a3,a5,a6 is a best choice: (−1)⋅(−2)⋅(−3)⋅2⋅(−1)=12.

In the third test case, choosing a1,a2,a3,a4,a5 is a best choice: (−1)⋅0⋅0⋅0⋅(−1)=0.

In the fourth test case, choosing a1,a2,a3,a4,a6 is a best choice: (−9)⋅(−7)⋅(−5)⋅(−3)⋅1=945.

解题思路
分别统计数字中正数、负数、0的个数。
容易发现，如果所有数字当中非0的数字一共少于5个，那么结果肯定是0，直接输出即可。
否则先将所有的数字从小到大排列。按照数字中负数的个数进行分类讨论，分为：负数有0或1个，负数有2或3个，负数有至少4个且存在非负数，只存在负数。这4种情况分别考虑。
另外可以发现，不论是哪种情况，数字中最大的那个数字肯定要选取在其中，不管它是正数、0还是最大的负数。

AC代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
long long a[100005];
int main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int t,n;
	cin>>t;
	while(t--)
	{
		cin>>n;
		int c1=0,c2=0,c3=0;
		for(int i=0;i<n;i++)
		{
			cin>>a[i];
			if(a[i]<0) c1++;
			if(a[i]==0) c2++;
			if(a[i]>0) c3++;
		}
		sort(a,a+n);
		if(c1+c3<5)
			cout<<"0"<<endl;
		else
		{
			if(c1==0||c1==1)
				cout<<a[n-1]*a[n-2]*a[n-3]*a[n-4]*a[n-5]<<endl;
			if(c1==2||c1==3)
			{
				if(a[0]*a[1]>a[n-4]*a[n-5])
					cout<<a[n-1]*a[n-2]*a[n-3]*a[0]*a[1]<<endl;
				else
					cout<<a[n-1]*a[n-2]*a[n-3]*a[n-4]*a[n-5]<<endl;
			}
			if(c1>=4&&c2+c3>0)
			{
				long long ans=a[n-1];
				if(a[0]*a[1]>a[n-4]*a[n-5])
					ans*=a[0]*a[1];
				else
					ans*=a[n-4]*a[n-5];
				if(a[2]*a[3]>a[n-2]*a[n-3])
					ans*=a[2]*a[3];
				else
					ans*=a[n-2]*a[n-3];
				cout<<ans<<endl;
			}
			if(c1>=5&&c2+c3==0)
				cout<<a[n-1]*a[n-2]*a[n-3]*a[n-4]*a[n-5]<<endl;
		}
	}
	return 0;
}