題目大意:
有一個天平,天平左右兩邊各有若干個鉤子,總共有C個鉤子,有G個鉤碼,求將鉤碼全部掛到鉤子上使天平平衡的方法的總數。
其中可以把天枰看做一個以x軸0點作為平衡點的橫軸
dp思路:
每向天平中方一個重物,天平的狀態就會改變,而這個狀態可以由若干前一狀態獲得。
首先定義一個平衡度j的概念:
當平衡度j=0時,說明天枰達到平衡,j>0,說明天枰傾向右邊(x軸右半軸),j<0則相反
那麼此時可以把平衡度j看做為衡量當前天枰狀態的一個值
因此可以定義一個 狀態數組dp[i][j],意為在掛滿前i個鉤碼時,平衡度為j的掛法的數量。
由於距離c[i]的范圍是-15~15,鉤碼重量的范圍是1~25,鉤碼數量最大是20
因此最極端的平衡度是所有物體都掛在最遠端,因此平衡度最大值為j=15*20*25=7500。原則上就應該有dp[ 1~20 ][-7500 ~ 7500 ]。
因此為了不讓下標出現負數,做一個處理,使使得數組開為 dp[1~20][0~15000],則當j=7500時天枰為平衡狀態
那麼每次掛上一個鉤碼後,對平衡狀態的影響因素就是每個鉤碼的 力臂
力臂=重量 *臂長 = w[i]*c[k]
那麼若在掛上第i個砝碼之前,天枰的平衡度為j
(換言之把前i-1個鉤碼全部掛上天枰後,天枰的平衡度為j)
則掛上第i個鉤碼後,即把前i個鉤碼全部掛上天枰後,天枰的平衡度 j=j+ w[i]*c[k]
其中c[k]為天枰上鉤子的位置,代表第i個鉤碼掛在不同位置會產生不同的平衡度
不難想到,假設 dp[i-1][j] 的值已知,設dp[i-1][j]=num(即已知把前i-1個鉤碼全部掛上天枰後得到狀態j的方法有num次)
那麼dp[i][ j+ w[i]*c[k] ] = dp[i-1][j] = num (即以此為前提,在第k個鉤子掛上第i個鉤碼後,得到狀態j+ w[i]*c[k]的方法也為num次
想到這裡,利用遞歸思想,不難得出 狀態方程dp[i][ j+ w[i]*c[k] ]= ∑(dp[i-1][j])
最終轉化為01背包問題
狀態方程dp[i][ j+ w[i]*c[k] ]= ∑(dp[i-1][j])
初始化:dp[0][7500] = 1; //不掛任何重物時天枰平衡,此為一個方法
復雜度O(C*G*15000)
#include<iostream> #include<cmath> #include<cstdio> #include<vector> #include<cstring> #include<algorithm> using namespace std; int dp[21][15005],w[25],c[20],n,m; int main() { int i,j,k; while(~scanf("%d%d",&n,&m)) { for(i=1;i<=n;i++) scanf("%d",&c[i]); for(i=1;i<=m;i++) scanf("%d",&w[i]); memset(dp,0,sizeof dp); dp[0][7500]=1; for(i=1;i<=m;i++)//m個物體逐個增加 { for(j=0;j<=15000;j++)//增加物體產生新的平衡度 要加上原有的平衡度 { if(dp[i-1][j])//優化 原有的平衡度是0的話 就不用加了 都一樣 { for(k=1;k<=n;k++)//將第i個物體放到第k個鉤上 dp[i][j+c[k]*w[i]]+=dp[i-1][j]; } } } printf("%d\n",dp[m][7500]); } return 0; }