P8669 [蓝桥杯 2018 省 B] 乘积最大

佚名 / 2024-02-26 / 原文

这是一开始的做法，死活有两个点过不去：

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string>
#include <cmath>
#define For(i, j, n) for (int i = j; i <= n; ++i)
using namespace std;
typedef long long LL;

const int N = 1e5 + 5;
const double INF = 1e10;
const LL mod = 1e9 + 9;

int n, k;
LL a[N];

bool cmp(int a, int b)
{
    return abs(a) > abs(b);
}

LL get_min()
{
    LL ans = 1;
    for (int i = n; i >= n - k + 1; i--)
    {
        ans = ans % mod * a[i] % mod;
    }
    return ans;
}

LL get_max(int b, int tar)
{
    LL ans = 1;
    for(int i = 1; i <= k; i++)
    {
        if(i != b)
        {
            ans = ans % mod * a[i] % mod;
        }
    }
    if(b != -1)
        ans = ans  % mod * a[tar] % mod;
    return ans;
}

int main()
{
    scanf("%d%d", &n, &k);
    for (int i = 1; i <= n; i++)
        scanf("%lld", &a[i]);
    sort(a + 1, a + n + 1, cmp);
    int cnt = 0, p1 = -1, p2 = -1, m1 = -1, m2 = -1;
    LL ans = 1;
    for (int i = 1; i <= k; i++)
        if (a[i] < 0)
            cnt++;
    if (cnt & 1) // 绝对值最大的k个数字只能得到负数
    {
        for (int i = k; i; i--)
        {
            if (p1 != -1 && m1 != -1)
                break;
            if (p1 == -1 && a[i] > 0)
                p1 = i;
            if (m1 == -1 && a[i] < 0)
                m1 = i;
        }
        for (int i = k + 1; i <= n; i++)
        {
            if (p2 != -1 && m2 != -1)
                break;
            if (p2 == -1 && a[i] > 0)
                p2 = i;
            if (m2 == -1 && a[i] < 0)
                m2 = i;
        }
        bool f1 = true, f2 = true;
        if(m1 != -1 && p2 != -1)
        {
            f1 = false;
            ans = get_max(m1, p2);
        }
        if(p1 != -1 && m2 != -1)
        {
            f2 = false;
            ans = max(ans, get_max(p1, m2));
        }
        if (f1 && f2) // 无法得到正数
        {
            ans = get_min();
        }
    }
    else
        ans = get_max(-1, -1);
    printf("%lld\n", ans);
    return 0;
}

这是贪心正解：

我们可以先考虑只有正数的情况，这时候只要选最大的数就行了；

而引入负数之后，因为负负得正，有可能会把原来最小的给反转成最大的，那么这时候，我们只要再把这个反转的可能性考虑进来即可，实现方式就是既考虑最大的那一头，也考虑最小的那一头。

#include<bits/stdc++.h>
#define int long long 
#define mod 1000000009
using namespace std;

int n,k,a[100005],ans;

signed main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++) cin>>a[i];
    sort(a+1,a+n+1);
    ans=k&1?a[n]:1;
    n-=k&1;
    k-=k&1;
    int l=1,r=n,f=ans<0?-1:1;
    while(k)
    {
        int p=a[l]*a[l+1],q=a[r]*a[r-1];//两端取
        if(p*f>q*f)//贪心选取
        {
            ans=p%mod*ans%mod;
            l+=2;
        }
        else
        {
            ans=q%mod*ans%mod;
            r-=2;
        }
        k-=2;
    }
    cout<<ans;
     return 0;
}