[LOJ6089]小 Y 的背包计数问题

题目

传送门 to LOJ

思路

不妨分成两部分讨论：$x<\sqrt{n}$ 和 $x\ge \sqrt{n}$ 。

为啥要这么分啊？这确实很难说。但是我们可以猜到，我们在让这两个值拿到平衡：值的种类和最多选取的个数。

当值的种类很少时，直接多重背包就行了。当选取的种类很少时呢？用 $f(x,i)$ 表示，选取 $i$ 个不小于 $\sqrt{n}$ 的数，使得其和为 $x$ ，数字的顺序不考虑。这个是很经典的问题，可以视作拼出一个阶梯型图案，一排一排的消除。所以有转移

$$f(x,i)=f(x-\sqrt{n},i-1)+f(x-i,i)$$

然后两边都是 $\mathcal{O}(n^{\frac{3}{2}})$ 的，拼在一起就可以了。

代码

#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long int_;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
inline int qkpow(int_ b,int_ q,int Mod){
	int ans = 1;
	for(; q; q>>=1,b=b*b%Mod)
		if(q&1) ans = ans*b%Mod;
	return ans;
}

const int MaxN = 100001;
const int Sqrt = 330;
const int Mod = 23333333;
int dp[Sqrt][MaxN]; // 背包
int f[Sqrt][MaxN]; // f[i][j]: j 分成 i 个数

int main(){
	int n = readint();
	dp[0][0] = 1;
	for(int i=1; i<Sqrt; ++i)
	for(int r=0; r<i; ++r) // r = j % i
	for(int j=r,s=0; j<=n; j+=i){
		s += dp[i-1][j]; // 求滑窗的和
		if(j >= i*(i+1))
			s -= dp[i-1][j-(i+1)*i];
		dp[i][j] = s = (s%Mod+Mod)%Mod;
	}
	f[0][0] = 1; // 啥子都没有
	for(int j=1; j<Sqrt; ++j)
	for(int i=j*Sqrt; i<=n; ++i)
		f[j][i] = (f[j-1][i-Sqrt]
			+ f[j][i-j]) % Mod;
	int ans = 0;
	for(int i=0; i<=n; ++i){
		int_ now = 0;
		for(int j=0; j<Sqrt; ++j)
			now = (now+f[j][i])%Mod;
		now *= dp[Sqrt-1][n-i];
		ans = (ans+now)%Mod;
	}
	printf("%d\n",ans);
	return 0;
}