递推DP HDOJ 5389 Zero Escape

题目传送门

 /*
     题意：把N个数分成两组，一组加起来是A，一组加起来是B，1<=A,B<=9，也可以全分到同一组。其中加是按照他给的规则加，就是一位一位加，超过一位数了再拆分成一位一位加。
     DP：dp[i][j]记录前i个数累加和为j的方案数，那么状态转移方程：dp[i][j+a[i]] += dp[i-1][j]; 当然，dp[i][a[i]] = 1;
         然后考虑几种特殊情况：都前往S1门或S2门，方案数+1。另外，比赛时我写出正确的转移方程，结果答案输出dp[n][s1]+dp[n][s2]将近是2倍，没多想，弃之，当时思绪很乱，以为方法错误。。。。
 */
 /************************************************
 * Author        :Running_Time
 * Created Time  :2015-8-13 14:39:58
 * File Name     :J.cpp
  ************************************************/

 #include <cstdio>
 #include <algorithm>
 #include <iostream>
 #include <sstream>
 #include <cstring>
 #include <cmath>
 #include <string>
 #include <vector>
 #include <queue>
 #include <deque>
 #include <stack>
 #include <list>
 #include <map>
 #include <set>
 #include <bitset>
 #include <cstdlib>
 #include <ctime>
 using namespace std;

 #define lson l, mid, rt << 1
 #define rson mid + 1, r, rt << 1 | 1
 typedef long long ll;
 const int MAXN = 1e5 + ;
 const int INF = 0x3f3f3f3f;
 const int MOD = ;
 int dp[MAXN][];
 int a[MAXN];
 int n, s1, s2;

 int cal(int x, int y)   {
     int ret = x + y;
     ret %= ;
     if (ret == )   return ;
     else    return ret;
 }

 int main(void)    {     //HDOJ 5389 Zero Escape
     int T;  scanf ("%d", &T);
     while (T--) {
         scanf ("%d%d%d", &n, &s1, &s2);
         int sum = ;
         for (int i=; i<=n; ++i)    {       //把N个数全加起来再按照规则处理和两个两个加是一样的
             scanf ("%d", &a[i]);    sum = cal (sum, a[i]);
         }

         memset (dp, , sizeof (dp));
         for (int i=; i<=n; ++i)    {
             dp[i][a[i]] = ;
             for (int j=; j>=; --j)    {
                 dp[i][j] = (dp[i][j] + dp[i-][j]) % MOD;
                 if (j + a[i] > )   {
                     int x = cal (j, a[i]);
                     dp[i][x] = (dp[i][x] + dp[i-][j]) % MOD;
                 }
                 else    {
                     dp[i][j+a[i]] = (dp[i][j+a[i]] + dp[i-][j]) % MOD;
                 }
             }
         }

         int ans = ;
         if (cal (s1, s2) == sum)    {       //两个门都能进去且条件符合
             ans += dp[n][s1];
             if (s1 == sum)  ans--;
         }
         if (s1 == sum)  ans++;      //可以全进一个门
         if (s2 == sum)  ans++;

         printf ("%d\n", ans);
     }

     return ;
 }