美团笔试其中一题——ABC三个数乘积最大化

最新推荐文章于 2024-08-29 22:25:44 发布

久未

最新推荐文章于 2024-08-29 22:25:44 发布

阅读量118

点赞数 3

文章标签：算法

本文链接：https://blog.csdn.net/weixin_44604409/article/details/141590474

版权

题目描述

给你三个正数abc和一个k，你有k次机会，每次可以将abc其中一个加一，求k次机会后，abc乘积的最大值
例如 a b c 为 1 2 1，k为2，则最大值为222 = 8

题解

思路：贪心，每次对最小的那个数+1，因此将输入的abc进行排序，使得a<=b<=c。
朴素模拟以下情况：

a < b 时
a == b < c 时
a == b == c 时

#include <iostream>
#include <string>
#include <set>
#include <unordered_set>
#include <unordered_map>
#include <vector>
#include <algorithm>
using namespace std;
using Type = unsigned long long;

const Type MOD = 1e9 + 7;

void just(Type &a, Type &b, Type &c)
{
    if (a > b)
        swap(a, b);
    if (a > c)
        swap(a, c);

    if (c < b)
        swap(c, b);

    a %= MOD, b %= MOD, c %= MOD;
}

Type mymin(Type a, Type b)
{
    if (a < b)
        return a;
    else
        return b;
}

int main()
{
    Type a, b, c, k;
    cin >> a >> b >> c >> k;
    just(a, b, c);
    while (1)
    {
        if (a == b && b == c)
        {
            if (k >= 3)
            {
                Type add = k / 3;
                a += add, b += add, c += add, k -= 3 * add;
            }
            else if (k == 2)
                a++, b++, k -= 2;
            else if (k == 1)
                a++, k--;
        }
        else if (a == b)
        {
            if (k >= 2)
            {
                Type n = k / 2;
                Type cha = c - b;
                Type add = mymin(n, cha);
                a += add, b += add, k -= 2 * add;
            }
            else
                a++, k--;
        }
        else if (a < b)
        {
            Type cha = b - a;
            Type add = mymin(k, cha);
            a += add;
            k -= add;
        }

        just(a, b, c);

        if (k <= 0)
            break;
    }

    cout << ((a * b) % MOD) * c % MOD << endl;

    return 0;
}