poj 3034 Whac-a-Mole

http://poj.org/problem?id=3034

题意：打地鼠游戏中，你有一个锤子，每一秒钟你可以拿着锤子移动d个单位的距离，掠过的鼠洞中露出的地鼠都会被锤打至，而事先知道从开始时各时间段内出现在老鼠的数量和位置，问题是从游戏开始至结束时，你最多能打到多少只地鼠，开始时锤子可以在任何位置。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int dp[35][35][35];
int c[35][35][35];
int n,d,m;

int gcd(int a,int b)
{
    return (a==0)?b:gcd(b%a,a);
}

int get(int x1,int y1,int x2,int y2,int t)
{
    int dx=x2-x1;
    int dy=y2-y1;
    if(dx==0&&dy==0) return c[x2][y2][t];
    int sum=0;
    if(dx==0)
    {
        if(y1>y2) swap(y1,y2);
        for(int i=y1; i<=y2; i++)
        {
            sum+=c[x2][i][t];
        }
        return sum;
    }
    if(dy==0)
    {
        if(x1>x2)swap(x1,x2);
        for(int i=x1; i<=x2; i++)
        {
            sum+=c[i][y1][t];
        }
        return sum;
    }
    int g=gcd(abs(dx),abs(dy));
    dx=dx/g;
    dy=dy/g;
    for(int i=0; i<=g; i++)
    {
        sum+=c[x1+i*dx][y1+i*dy][t];
    }
    return sum;
}

int main()
{
    while(scanf("%d%d%d",&n,&d,&m)!=EOF)
    {
        int max1=0;
        memset(dp,0,sizeof(dp));
        memset(c,0,sizeof(c));
        if(n==0&&d==0&&m==0) break;
        for(int i=0; i<m; i++)
        {
            int x,y,t1;
            scanf("%d%d%d",&x,&y,&t1);
            max1=max(max1,t1);
            c[x+d][y+d][t1]=1;
        }
        n=n+2*d;
        for(int t=1; t<=max1; t++)
        {
            for(int i=0; i<n; i++)
            {
                for(int j=0; j<n; j++)
                {
                    for(int x2=0; x2<n ; x2++)
                    {
                        for(int y2=0; y2<n; y2++)
                        {
                            if((i-x2)*(i-x2)+(j-y2)*(j-y2)<=d*d)
                            {
                                int ans=get(x2,y2,i,j,t);
                                dp[t][i][j]=max(dp[t][i][j],dp[t-1][x2][y2]+ans);
                            }
                        }
                    }
                }
            }
        }
        int max2=0;
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
                max2=max(max2,dp[max1][i][j]);
            }
        }
        printf("%d
",max2);
    }
    return 0;
}