LOJ#3367. 「IOI2020」装饼干

Description

• 将$k$种饼干放入$x$个袋子之中，使得每一个袋子的饼干价值相同，求有多少种不同的价值能够满足。
• 第$i$个饼干有$a_i$个，价值为$2^{i-1}$.
• $k\le 60,\sum a_i\le 1e18$

Solution

• 很容易想到，贪心放饼干，从高位往低位放，尽量把大的饼干塞入袋子里面，这样一定不劣。
• 于是就有了一个简单的DP思路，设$f[i][j]$表示放到第$i$位，当前小于$i$的剩下的饼干的和为$j$。
• 转移的状态数乍一看感觉很多，实际上仔细观察一下：
  $$f[i][j]\to f[i-1][min(j-x{2}^{i-1},s_{i-2},2^{i-1}-1)]$$
  $$f[i][j]\to f[i-1][min(j,s_{i-2},2^{i-1}-1)]$$
• 这种转移的状态实际上是有限的 ，理解上并不太直观。
• 或者可以考虑另一种方法：
• 如何判定一个$y$是否满足条件，从低位往高位做，容易得到
  $$[\frac{su{m}_{i-1}}{x}]\ge y\%2^{i+1}$$
• 那么假设$li{m}_i=[\frac{su{m}_{i-1}}{x}]$，从高位往低位考虑$y$的取值，如果当前第$i$位$y$与$lim$相同，那么继续递归下去，否则$y<lim$，那么限制就转化为了只有$i-1$的限制，直接从$f_{i-1}$转移过来，因此只有$k^2$种状态。

#include "biscuits.h"
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<map>
#define ll long long 
#define maxn 63
using namespace std;

map<ll,ll> f[maxn];
ll sum[maxn],X,_2[maxn];
ll dp(int n,ll s){
	if (s/X>=_2[n]-1) s=(_2[n]-1)*X;
	if (f[n].find(s)!=f[n].end()) 	
		return f[n][s];
	if (n==0) return f[n][s]=1;
	ll &tmp=f[n][s];
	f[n][s]=dp(n-1,min(s,sum[n-1]));
	if (s/_2[n-1]>=X)
		f[n][s]+=dp(n-1,min(s-X*_2[n-1],sum[n-1]));
	return f[n][s];
}

ll count_tastiness(ll x, std::vector<ll> a){
	int i,j,k; X=x;
	for(i=0;i<=61;i++) f[i].clear();
	_2[0]=1;for(i=1;i<=61;i++) _2[i]=_2[i-1]<<1;
	for(i=0;i<a.size();i++) 
		sum[i+1]=sum[i]+a[i]*_2[i];
	for(i=a.size();i<=60;i++) sum[i+1]=sum[i];
	return dp(61,sum[a.size()]);
}