求 k 整除最大元素和(dp)

Description

复制代码
给你一个整数数组,请你在其中选取若干个元素,
使得其和值能被 k 整除,输出和值最大的那个和值。
最后的数字可能很大,所以结果需要对 19260817 取模。

Input

复制代码
第一行是两个正整数 n,k:表示数组的长度,以及被整除的除数 k。
接下来是 n 行,每行是一个正整数 num_i,表示数组中第 i 个数。
n <= 10^5,  k <= 100, num_i <= 10^9。

Output

复制代码
能被 k 整除的元素最大和。

Sample Input

复制代码
5 3
3
5
1
8
6

Sample Output

复制代码
18

思路:

将n个数取余分到0-(k-1)数组内,然后dp,dp[i][j]代表前0-i内的数相加,余数为j的最大值。

#define _CRT_SECURE_NO_WARNINGS

#include<iostream>

#include<string>

#include<cstring>

#include<cmath>

#include<ctime>

#include<algorithm>

#include<utility>

#include<stack>

#include<queue>

#include<vector>

#include<set>

#include<math.h>

#include<map>

#include<unordered_map>

using namespace std;

typedef long long LL;

typedef unsigned long long ULL;

const int N = 1000;

LL dp[110][110];

vector<LL> p[110];

LL n,k,x;

bool cmp(LL x, LL y)

{

return x > y;

}

int main() {

cin >> n >> k;

for (int i = 1; i <= n; i++)

{

scanf("%lld", &x);

p[x % k].push_back(x);

}

for (int i = 0; i <= k-1; i++)

sort(p[i].begin(), p[i].end(), cmp);

x = 0;

for (int i = 0; i <p[0].size(); i++)

x += p[0][i];

dp[0][0] = x;

for (int i = 1; i <= k - 1; i++)

{

LL sum = 0;

for (int j = 0; j < p[i].size(); j++)

{

sum += p[i][j];

x = (j + 1) * i % k;

for (int w = 0; w <= k - 1; w++)

{

if (j == 0) dp[i][w] = max(dp[i - 1][w], dp[i][w]);

if (dp[i - 1][w])

dp[i][(x + w) % k] = max(dp[i][(x + w) % k], dp[i - 1][w] + sum);

}

}

}

cout << dp[k-1][0] % 19260817 << endl;

return 0;

}

相关推荐
放下华子我只抽RuiKe53 小时前
算法的试金石:模型训练、评估与调优的艺术
人工智能·深度学习·算法·机器学习·自然语言处理·数据挖掘·线性回归
oem1103 小时前
C++中的享元模式实战
开发语言·c++·算法
流云鹤3 小时前
每日一题0316
算法
leonkay4 小时前
Golang语言闭包完全指南
开发语言·数据结构·后端·算法·架构·golang
颜酱4 小时前
BFS 与并查集实战总结:从基础框架到刷题落地
javascript·后端·算法
casual~5 小时前
第?个质数(埃氏筛算法)
数据结构·c++·算法
仰泳的熊猫5 小时前
题目2308:蓝桥杯2019年第十届省赛真题-旋转
数据结构·c++·算法·蓝桥杯
hssfscv5 小时前
力扣练习训练2(java)——二叉树的中序遍历、对称二叉树、二叉树的最大深度、买卖股票的最佳时机
java·数据结构·算法
y = xⁿ6 小时前
【LeetCodehot100】二叉树大合集 T94:二叉树的中序遍历 T104:二叉树的最大深度 T226:翻转二叉树 T101:对称二叉树
后端·算法·深度优先
不想看见4046 小时前
Search a 2D Matrix II数组--力扣101算法题解笔记
数据结构·算法