1. 程式人生 > >51nod 1597 有限揹包計數問題 DP 根號分治

51nod 1597 有限揹包計數問題 DP 根號分治

題解:

考慮根號分治。
對於體積 n \le\sqrt n 的東西,發現揹包可以用一個字首和優化。
對於體積 > n

>\sqrt n 的東西,實際上每個物品都可以看做有無限個 ,就是求把某個數分成若干份,每份都至少為 n + 1 \sqrt n+1
的方案數,由於分成的份數不會超過 n \sqrt n ,所以用一個簡單的DP就行了。

程式碼:

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=100010; const int inf=2147483647; const int mod=23333333; int read() { int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar(); return x*f; } int n,f[Maxn],g[2][Maxn],s[Maxn],ff[Maxn]; void upd(int &x,int y){x+=y;if(x>=mod)x-=mod;} int main() { memset(ff,0,sizeof(ff)); memset(f,0,sizeof(f)); n=read();int m=(int)sqrt(n); f[0]=1; for(int i=0;i<=n;i++)s[i]=1; for(int i=1;i<=m;i++) { for(int j=n;j>=i;j--) { int k=min(i,j/i); upd(f[j],(s[j-i]-((j-(k+1)*i>=0)?s[j-(k+1)*i]:0)+mod)%mod); } s[0]=1; for(int j=1;j<=n;j++)s[j]=(f[j]+((j>=i+1)?s[j-(i+1)]:0))%mod; } int now=0,ans=0; memset(g[now],0,sizeof(now)); g[0][0]=1; for(int i=1;i<=m;i++) { now^=1; memset(g[now],0,sizeof(g[now])); for(int j=i;j<=n-m*i;j++) { upd(g[now][j],g[now^1][j-1]),upd(g[now][j],g[now][j-i]); upd(ans,(LL)g[now][j]*f[n-i*m-j]%mod); } } upd(ans,f[n]); printf("%d",ans); }