[WC2022-DMY]Robbery

题目

传送门 to usOJ

题目描述
有 $n$ 种物品，第 $i(1⩽i⩽n)$ 种物品质量为 $i$，价值为 $a_i$，数量无限。

给定 $k,w$，请选择 $k$ 个物品，质量和为 $w$ 且价值之和最大。输出这个价值和。

数据范围与提示
$n⩽10^3$ 但 $k⩽10^6$ 。复杂度与 $w\in [k,kn]$ 和 $a_i\in [1,10^9]$ 无关。

思路

感觉这个做法可以称得上 $\text{constructive}$ 了。

猫腻在于：质量很小，且质量和需要恰好达到 $w$ 。考虑到 $k⩽10^6$，如果往 $\log k$ 的方向思考，说不定就会醍醐灌顶！

定理：若 $2\mid k$，则可以将一个方案 划分成两组 同样多的物品，且两组的质量和的差值不超过 $n$ 。

证明：设二者质量和的差值超过 $n$ 。则质量和较大的那一堆的 $\max >$ 质量和较小的那一堆的 $\min$ 。二者交换，将会带来一个 $2n$ 以内的变化量。而原本差值超过 $n$，减去这个 $2n$ 以内的变化量后，得到的差值的绝对值比原来更小。不断调整即可。

于是，若我们需要求出 $k$ 个物品、质量和在 $[a,b]$ 内的答案，只需要递归 $\frac{k}{2}$ 个物品、质量和在 $[\frac{a-n}{2},\frac{b+n}{2}]$ 内的答案。而 $2\nmid k$ 时，就暴力递归 $(k-1)$ 个物品、质量和在 $[a-n,b]$ 内的答案。复杂度是多少呢？

注意到上面有 $÷2$ 的操作，我们大胆猜测区间长度有上界。最坏情况是 $2\nmid k$，此时区间长度会先加 $n$，再 $÷2$ 后加 $n$ 。解方程 $L=\frac{L+n}{2}+n$ 得 $L=3n$，即，区间长度为 $3n$ 就不会再增加了！所以背包合并的复杂度稳定在 $\mathcal{O}(n^2)$，总复杂度 $\mathcal{O}(n^2\log k)$ 。

代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cctype>
using namespace std;
# define rep(i,a,b) for(int i=(a); i<=(b); ++i)
# define drep(i,a,b) for(int i=(a); i>=(b); --i)
typedef long long llong;
inline int readint(){
	int a = 0, c = getchar(), f = 1;
	for(; !isdigit(c); c=getchar())
		if(c == '-') f = -f;
	for(; isdigit(c); c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
template < typename T >
inline void getMax(T &a,const T &b){
	if(a < b) a = b;
}

const int MAXN = 1003, MAXM = MAXN*3;
const long long INF = (1ll<<62)-1;
int a[MAXN], n, me;
llong dp[2][MAXM];
void solve(int k,int l,int r){
	if(k == 1){
		for(int i=l; i<=r&&i<=n; ++i)
			dp[me][i-l] = a[i]; // from int to llong
		if(!l) dp[me][0] = -INF; // not defined
		rep(i,n+1,r) dp[me][i-l] = -INF;
	}
	else if(k&1){
		solve(k-1,l-n,r-1), me ^= 1;
		fill(dp[me],dp[me]+r-l+1,-INF);
		rep(i,l-n,r-1) rep(j,max(1,l-i),n){
			if(i+j > r) break; // too big j
			getMax(dp[me][i+j-l],dp[me^1][i-l+n]+a[j]);
		}
	}
	else{
		const int low = (l-n+1)>>1;
		solve(k>>1,low,(r+n)>>1);
		fill(dp[me^1],dp[me^1]+r-l+1,-INF);
		rep(i,low,(r+n)>>1) rep(j,max(l-i,low),(r+n)>>1){
			if(i+j > r) break; // too big j
			getMax(dp[me^1][i+j-l],dp[me][i-low]+dp[me][j-low]);
		}
		me ^= 1; // who am I
	}
}

int main(){
	n = readint();
	int k = readint(), w = readint();
	rep(i,1,n) a[i] = readint();
	solve(k,w,w);
	printf("%lld\n",dp[me][0]);
	return 0;
}

后记

质量和固定，其实可以联想背包合并的。有点像这道题，又不太像。

质量很小，则是为了使得分治可行。也比较像倍增。像贪心，但硬要说，则又说不出个门道来。说是这种部分贪心吧，还是不能算。真奇怪啊。