poj 1837 DP

来源：互联网发布：大疆飞控算法编辑：程序博客网时间：2024/06/04 18:55

dp【i】【j】表示加入第i个物品之后和力矩为j的个数

由于数组下标不能为负数，所以全部加上7500（15*25*20）

AC代码如下：

#include <iostream>#include <cstring>#include <cstdio>#include <algorithm>using namespace std;int dp[21][16000];int pos[21], weight[21];int C, G;int main(){    while( scanf( "%d%d", &C, &G ) != EOF ){        for( int i = 1; i <= C; i++ ){            cin >> pos[i];        }        for( int i = 1; i <= G; i++ ){            cin >> weight[i];        }        memset( dp, 0, sizeof( dp ) );        dp[0][7500] = 1;        for( int i = 1; i <= G; i++ ){            for( int j = 1; j <= C; j++ ){                int temp = weight[i] * pos[j];                if( temp >= 0 ){                    for( int k = temp; k <= 15000; k++ ){                        dp[i][k] += dp[i-1][k-temp];                    }                }else{                    for( int k = 15000 + temp; k >= 0; k-- ){                        dp[i][k] += dp[i-1][k-temp];                    }                }            }        }        cout << dp[G][7500] << endl;    }    return 0;}

0 0