多重背包方案数的求解

原函数求解多重背包方案数

原题：找单词

题意：

有x1个字母A， x2个字母B,….. x26个字母Z，同时假设字母A的价值为1，字母B的价值为2,….. 字母Z的价值为26。那么，对于给定的字母，可以找到多少价值<=50的单词（ABC和ACB视为一个单词）。

分析：

拿出k个单词价值i，可以使dp[j]转化为dp[j+k*i]（dp[i]表示价值和为i的方案数），比如有3个B(价值2)，所有价值为6加上三个B就可以变成价值为12的。

如果不开两个数组，从小开始加，加3个B的时候，前面有些累计加的也是3个B（一个加两个）会重复，所以开两个数组排除干扰。

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
#include<list>
#include<vector>
#include<stack>
#include<queue>
#include<ctime>
#include<cstdlib>
#include<sstream>
//#include<windows.h> 
#include<functional>
#define D long long
#define F double
#define MAX 0x7fffffff
#define MIN -0x7fffffff
#define mmm(a,b) memset(a,b,sizeof(a))
#define pb push_back  
#define mk make_pair  
#define fi first  
#define se second  
#define pill pair<int, int>  
#define for1(i,a,b) for(int i=a;i<=b;i++)
#define for2(i,a,b) for(int i=a;i>=b;i--)
#define ini(n) scanf("%d",&n)
#define inll(n) scanf("%lld",&n)
#define outisp(n) printf("%d ",n)
#define outllsp(n) printf("%lld ",n)
#define outiel(n) printf("%d\n",n)
#define outllel(n) printf("%lld\n",n)
using namespace std; 
#define N 500100 
#define MOD ((int)1e9+7)
#define random(a,b) (rand()%(b-a+1)+a)
#define stop Sleep(2000)
#define CLS system("cls")
const string el="\n";
const string elel="\n\n";
const string sp=" ";
const string spsp="  ";
const string tab="\t";

D dp[100];
int x[27];
D tmp[100];

int main(){
    int t;ini(t);while(t--){

        mmm(dp,0);mmm(tmp,0);
        for1(i,1,26)ini(x[i]);

        for1(i,0,x[1])dp[i]=tmp[i]=1;

        //for1(i,0,50)outllsp(dp[i]);cout<<el;

        for1(i,2,26){
            for1(k,1,x[i]){
                for1(j,k*i,50){
                    dp[j]+=tmp[j-k*i];
                }
            }
            for1(j,0,50)tmp[j]=dp[j];


        //for1(i,0,50)outllsp(dp[i]);cout<<el;
        }
        D ans=0;
        for1(i,1,50)ans+=dp[i];
        cout<<ans<<el;
    }
} 

特点题目的原函数优化

原题：比昨天更多的棒棒糖 (Hard)

题意：

有n元，每天花一定数量，要求每天都不能比昨天花的少（>=），第一天至少花x元，不能连续k天花相同数量的钱。求有多少种花法。（不要求花完，即花了一定数量后可以结束或者继续花）

1. 用多重背包原函数的思路看，大小为n的背包，有大小为x~n这么几种物品，每种物品有k-1件，求一共有几种装法，复杂度O(k*n^2)，最坏情况1e12。。。
2. 用二进制优化多重背包后，在算方案数时会出错，eg：k=6，分成1,2,2；即挑选2个时的所有方案都会变成原来的2倍。。。（当然最大值时这种办法还是很好的）
3. 其实这题有个特点，所有包的数量都相同，所以稍加改进就可以转化为01包+完全包。
4. 强调：只可以在包数量相同时使用

先附上代码：

#include <iostream>  
#include <stdio.h>  
#include <time.h>  
#include <string.h>  
#include <algorithm>  
#define llt long long  
#define Mod 998244353  
using namespace std;  
const string el="\n";


llt F[10001];  

int main(){  

    int n,x,k;  
    cin>>n>>x>>k;  
    memset(F,0,sizeof(F));  
    F[0]=1;  
    for(int i=x;i<=n;++i){
        cout<<"when i is "<<i<<el; 
        for(int j=i; j<=n; ++j)//完全背包  
            F[j] = (F[j] + F[j-i]) % Mod;

        for(int j=1; j<=n; ++j)printf("%2lld ",F[j]);
        cout << el;

        for(int j=n; j>=k*i; --j)//0-1背包  
            F[j] = (F[j] - F[j-k*i] + Mod) % Mod;

        for(int j=1; j<=n; ++j)printf("%2lld ",F[j]);
        cout << el;  
    }  

    llt ans=0;  
    for(int i=1;i<=n;++i)  
        ans=(ans+F[i]+Mod)%Mod;  
    cout<<ans<<endl;  
    return 0;  
}  

下面是测试数据打表

解析：

F[i]代表符合题意的花i元的方式的种数。
图片中，对于每个i的第一排是完全背包后的数组，第二排是01后的数组。

先看i==2时，在没有k的限制的情况下，花2元是一种（第一天至少2元），花4元也是一种（因为i才刚开始循环，所以不考虑第一天就4元的情况，即花钱数只能为2），花2的倍数都是一种。就是第一排展现的那样。

完全背包的实际上翻译过来就是：对于每个数j，都可以通过花了j-i的花法，再花i元来得到，即 F[j] = (F[j] + F[j-i]) % Mod;

而第二排是使用k来限制，对于i*k以上的数 j，在完全包中都加了一次F[j-i*k]（因为i是从小到大循环，所以对于i*k以上的数不可能是比i大的数形成的，而由比i小的形成的那部分又是F[j-i*k]中不包含的，所以突破k的限制的那部分就是F[j-i*k]），所以F[j] = (F[j] - F[j-k*i] + Mod) % Mod。