蓝桥省赛 倍数问题 DP 背包 贪心

🍑 算法题解专栏

---

🍑 倍数问题

输入

4 3
1 2 3 4

输出

9

---

🍑 三维DP

🍤 状态定义 f[ i ][ j ][ k ]
🍁 所有从前 i 个数种选，且已经选了 j 个数 总和模K的余数为 k 所有方案的集合
🍁 每一维都是限制

🍤 状态转移
🍁 ① 选第 i 个物品
🍁 ② 不选第 i 个物品

 f[i][j][k]=max(f[i-1][j][k], f[i-1][j-1][((k-a[i])%m+m)%m]+a[i]);

🍑 二维DP（贪心优化）（每种余数的数只处理前 3 大的数）

🍤 java定义List数组：static ArrayList[] a = new ArrayList[N];
🍤 初始化：a[i] = new ArrayList<>();

---

🍑 状态转移
🍁 在前 i 个数种选，已经选了 j 个数，并且余数为 k 所有方案的集合的总和最大值

🥞 三维
f[cnt][j][k]=max(f[cnt-1][j][k],f[cnt-1][j-1][((k-v)%m+m)%m]+v);

🥞 二维（滚动数组优化）
f[j][k] = Math.max(f[j][k], f[j - 1][(k - x % m + m) % m] + x);

---

import java.util.*;

public class Main
{
	static int n, m, N = 1010, INF = 0x3f3f3f3f;// m 代码 MOD
	static int[][] f = new int[4][N];// i 表示已选数的个数，j 表示余数
	static ArrayList[] a = new ArrayList[N];// 余数集合数组

	public static void main(String[] args)
	{
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		m = sc.nextInt();

//		初始化余数表
		for (int i = 0; i < N; i++)
			a[i] = new ArrayList<>();

//		输入数据
		for (int i = 0; i < n; i++)
		{
			int x = sc.nextInt();
			a[x % m].add(x);
		}

//		初始化
		for (int i = 0; i <= 3; i++)
			Arrays.fill(f[i], -INF);
		f[0][0] = 0;

		for (int i = 0; i < m; i++)
		{
			Collections.sort(a[i]);// 对每一个余数集合进行排序
			Collections.reverse(a[i]);// 默认排升序 倒过来即降序
//选择前 3 大（答案的三个数可能会 同余，所以至少要选 3 个） min操作是防止该余数对应的数少于三个
			for (int u = 0; u < Math.min(3, a[i].size()); u++)
			{
				int x = (int) a[i].get(u);//前3大的数取出处理（其他的都跳过）
				for (int j = 3; j >= 1; j--)//枚举选了多少个数
				{
					for (int k = 0; k < m; k++)//枚举余数
					//状态转移方程与三维一致，只是少了一维
						f[j][k] = Math.max(f[j][k], f[j - 1][(k - x % m + m) % m] + x);
				}
			}
		}
		System.out.println(f[3][0]);
	}
}

👨‍🏫 大佬题解