CodeForce 505C Mr. Kitayuta(DP+DFS)

http://codeforces.com/contest/505/problem/C

题意是：有30001个岛屿，有的岛屿上存在宝石。给出岛屿上宝石的情况和第一次跳的步数。并且这个人每次跳的步数是 上次跳的步数/上次跳的步数 - 1/ 上次跳的步数 +1。如果跳到的岛屿上有宝石，那么将获得这个岛屿上宝石的数量。

用DP+dfs来做。dp[i][j]表示跳到长度为i的岛屿，j表示当前跳跃的步数。
很明显，O(30000*30000)是不可取的。
经过仔细思考可以发现每次跑一次程序，他最多可以跳跃的步数只有250左右。可以这样想，当第一步跳的时候可以跳跃的长度为1，那么下面每次跳都在当前增加一个长度的时候，构成n*(n+1)<=30000,解出n在250左右。所以j的状态最多可以有250种。
注意状态最多有250种并不是状态是 1,2,3,4，….250 这250种。而是以长度d开始的250种状态。所以开一个dp[30000][250*2]的数组，将每次跳的长度映射到里面。也就是说将 d->250,d+1->251….

#include <bits/stdc++.h>
using namespace std;

const int N = 30000+10;
int dp[N][2*250];
int vis[N][2*250];
int gem[N];
int maxn=-1,xx;
int dfs(int i,int j)
{//j表示实际的长度 jj表示映射的长度
     int jj = j - xx;
     if(i>maxn) return 0;//不知道为什么改成return dp[i][jj]就RE
     if(vis[i][jj]) return dp[i][jj];//不太明显的剪枝

     vis[i][jj] = true;
     if(j<=1) dp[i][jj] = gem[i]+max(dfs(i+j,j),dfs(i+j+1,j+1));
     else dp[i][jj] = gem[i]+max(dfs(i+j,j),max(dfs(i+j-1,j-1),dfs(i+j+1,j+1)));

     return dp[i][jj];
}
int main()
{
    memset(dp,0,sizeof(dp));
    memset(gem,0,sizeof(gem));
    memset(vis,false,sizeof(vis));
    int n,d;
    scanf("%d%d",&n,&d);
    while(n--)
    {
        int x;
        scanf("%d",&x);
        maxn = max(maxn,x);
        gem[x]++;
    }
    xx = d - 250;//映射差值
    printf("%d\n",dfs(d,d));
    return 0;
}