[hdoj]1244记忆化搜索

 1 #include <iostream>
 2 using namespace std;
 3 
 4 const int N = 1005;
 5 const int M = 25;
 6 int ans[N][M],segs[M],ints[N];
 7 int n,m;
 8 
 9 int DFS(int posInt,int posSeg)
10 {
11     if (ans[posInt][posSeg] != -1)
12         return ans[posInt][posSeg];
13     //已经用完了n个数 或 分完m个数段了
14     if (posSeg > m || posInt > n)
15     {
16         ans[posInt][posSeg] = 0;
17         return ans[posInt][posSeg];
18     }
19     //判断，即使取了posInt位置的数，能不能取完余下的数段
20     int remainedInts = n - posInt + 1;
21     int remainedSegs = 0;
22     for (int i = posSeg;i <= m;i ++)
23         remainedSegs += segs[i];
24     //余下的数段的长度之和比余下的数字的个数还大
25     if (remainedSegs > remainedInts)
26     {
27         ans[posInt][posSeg] = 0;
28         return ans[posInt][posSeg];
29     }
30     //若能走到这一步，说明posInt位置的数可以不作为第posSeg个数段的起始数字
31     int max;
32     //以第posInt个数为起始，取seg[posSeg]个连续的数的和
33     int sum = 0;
34     for (int i = posInt;i < posInt + segs[posSeg];i ++)
35         sum += ints[i];
36     max = sum + DFS(posInt + segs[posSeg],posSeg + 1);
37     //以第posInt + 1个数为起始，取seg[posSeg]个连续的数的和
38     int temp = DFS(posInt + 1,posSeg);
39     max = max > temp ? max : temp;
40     ans[posInt][posSeg] = max;
41     return ans[posInt][posSeg];
42 }
43 
44 int main ()
45 {
46     while (scanf("%d",&n) != -1 && n != 0)
47     {
48         memset(ans,-1,sizeof(ans));
49         scanf("%d",&m);
50         for (int i = 1;i <= m;i ++)
51             scanf("%d",&segs[i]);
52         for (int i = 1;i <= n;i ++)
53             scanf("%d",&ints[i]);
54         printf("%d\n",DFS(1,1));
55     }
56     return 0;
57 }