[Acwing|蓝桥杯] 1234. 倍数问题 背包dp+优化

前言

传送门 :

思路

看完题目之后大概就知道是 动态规划了

状态表示 : $dp[i][j][k]$从前$i$个数中选$j$个总和正好是$k$的最大值

因为要求的是 $k$的倍数,所以我们对第三层取$\%$ ,即总和正好是$\%k$的最大值

状态计算 :
$dp[i][j][k]=max(dp[i-1][j][k],dp[i-1][j-1][((k-a[i])\%K+K)\%K]+a[i]$)

$((k-a[i]\%K)+K)\%K$这一个是将负数取模转正,很常用了属于是

但是如果这么搞上去,会$MLE$,$1e5\ast 1e3\ast 3$,(题解的那个$vector$也$MLE$了)

因此我们考虑同背包问题一样优化第一维

因此状态转移变为 :
只不过第二维度的循环需要倒序
$dp[j][k]=max(dp[j][k],dp[j-1][((k-a[i])\%K+K)\%K]+a[i]$)

但是处理完$MLE$之后,又会遇到$TLE$的问题

因为我们是,三层循环的进行转移$O(N\ast K)$

for(int i=1;i<=n;i++)
	for(int j=3;j>=1;j--)
		for(int k=0;k<K;k++)

显然第二层,第三层无法优化,我们考虑优化第一层

我们考虑将所有数 按照$\%k$分类

例如 $7\%5=2,2\%5=2,[2,7]$

因为相同余数 和 其他数相加之后再求余都是一样的

因为题目中需要求的是最大 , 但是我们不能简单的拿每个类别中的最大而已

因为存在 三个都是 $\%k$相同的 , 因此我们对于每一个类别,拿前三个最大

Mycode

const int N = 1e5 + 10, M = 1e3+10;

// int dp[N][4][M];
int f[5][M];

int n,K;
int a[N];

vector<int> allr[N];

void solve(){
	cin>>n>>K;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		allr[a[i]%K].pb(a[i]);
	}

	memset(f,-0x3f,sizeof f);
	f[0][0] = 0 ;

	for(int i=0;i<K;i++){
		sort(all(allr[i]),greater<int>());
		for(int u = 0 ;u<3 && u< allr[i].size() ; u ++ ){
			int t = allr[i][u];
			for(int j=3;j>=1;j -- ){
				for(int k = 0;k<K;k++)
				f[j][k] = max(f[j][k],f[j-1][((k-t)%K+K)%K]+t);
			}
		}
	}

	cout<<f[3][0]<<endl;

}