蓝桥杯 20. 倍数问题

倍数问题

原题目链接

题目描述

众所周知,小葱同学擅长计算,尤其擅长判断一个数是否是另一个数的倍数。但当面对多个数时,他就比较苦恼了。

现在小葱给了你 n 个数,希望你从中找出三个数,使得这三个数的 和是 K 的倍数 ,并且这个 和最大。

题目保证一定存在解。

输入描述

  • 第一行包含两个正整数 n 和 K。
  • 第二行包含 n 个正整数,代表给定的数列。

数据范围:

  • 1 ≤ n ≤ 10⁵
  • 1 ≤ K ≤ 10³
  • 所有给定的整数不超过 10⁸

输出描述

输出一行一个整数,表示满足条件的最大和。

输入示例

in 复制代码
4 3
1 2 3 4

输出示例

out 复制代码
9

c++代码

cpp 复制代码
#include<bits/stdc++.h>

using namespace std;

typedef long long ll;

int main() {
    ll n, K, x, a, b, c, ans = 0;
    cin >> n >> K;
    vector<vector<ll>> arr(K);
    for (ll i = 0; i < n; i++) cin >> x, arr[x % K].push_back(x);
    for (ll i = 0; i < K; i++) sort(arr[i].begin(), arr[i].end());
    for (ll i = 0; i < K; i++) {
        if (arr[i].size() <= 0) continue;
        a = arr[i].back(), arr[i].pop_back();
        for (ll j = i; j < K; j++) {
            if (arr[j].size() <= 0) continue;
            b = arr[j].back(), arr[j].pop_back();
            ll val = K - i - j;
            while(val < 0) val += K;
            while(val <= K - 1) {
                if (arr[val].size() > 0) c = arr[val].back(), ans = max(ans, a + b + c);
                val += K;
            }
            arr[j].push_back(b);
        }
        arr[i].push_back(a);
    }
    cout << ans;
    return 0;
}//by wqs

题目解析

给你 n 个数和一个整数 K,从这 n 个数中选出三个数 a, b, c,要求:

  • (a + b + c) % K == 0
  • a + b + c 最大化

假设(a + b + c) % K = 0,

那么(a % K + b % K + c % K) % K = 0,

假设a % K = d, b % K = e,

根据(d+ e + c % K) % K = 0,并且数据的值都是正数

也就是说d + e + c % K = n * K(n >= 1),

也就是说c % K = n * K - d - e

那么c % K = K - d - e 或者 2 * K - d - e或者3 * K - d - e...

不要认为有很多种情况,我们看看c % K的范围,

显然,0 <= c % K <= K - 1,

也就是0 <= n * K - d - e <= K - 1,

可见在这个范围下不会有多少种情况,因为K的最大值才1000。

算法流程

枚举两个数的余数,然后计算第三个数所需的余数,根据贪心算法并在对应余数组中取最大值。

1. 分类存储:

将所有数字按其对 K 取模的结果分成 K 个桶,即: arr[i] 存放所有 x 满足 x % K == i。

这样做的目的,是为了快速找出所有具有相同模值的数,为后续组合提供便利。

2. 排序每个桶:

为了后续能快速取出某个模值的最大值,对每个桶排序 ,这样能轻松获取每个模值中最大的元素。

3. 枚举前两个数模值 (i, j):

你遍历所有 i, j ∈ [0, K),对每一组 (i, j):

  • 从 arr[i] 取最大值 a
  • 从 arr[j] 取最大值 b

4. 根据前两个快速得出第三个模值k:

k = n * K - i - j(n >= 1, 0 <= k <= K - 1)

5. 特别处理下标冲突:

你保存了桶中最大的数后还要回填,因为之后还会用到这些桶。每次取出最大值后记得放回去。

相关推荐
机器人零零壹41 分钟前
技术白皮书:iRobotCAM如何解决机器人激光切割应用中的核心难点,从CAD离线编程到双目视觉定位与轨迹自适应提升复杂空间曲线切割的精度、效率与柔性化水平
人工智能·算法
ZhengEnCi3 小时前
大模型的分类到底怎么分:从基础模型到推理模型、从通用到专用的完整图谱
算法
取名真是4 小时前
Python 算法知识点从基础到进阶
开发语言·python·算法
罗西的思考5 小时前
【轻量化复刻大模型 / KIMI K3】KDA 模块交互式演示
人工智能·算法·机器学习
nm_zfyjt7 小时前
信息流广告投放优化实践:线索成本评估模型与转化链路重构(家装建材案例)
算法·重构·内蒙古朋友圈广告服务商·内蒙古朋友圈获客
w_zero_one8 小时前
算法--构造
java·数据结构·算法
其实防守也摸鱼8 小时前
信创安全竞赛知识总结与题库
大数据·网络·人工智能·算法·安全·php
小超人诞生日记10 小时前
C++ 智能指针的使用及其原理详解
c++·算法
AiNightVision13 小时前
无人机夜视模块如何选择?——从参数指标到系统集成的完整选型指南
人工智能·算法·计算机视觉·车载系统·无人机·视频·智能硬件