poj 3034 Whac-a-Mole

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 using namespace std;
 5 
 6 int dp[35][35][35];
 7 int c[35][35][35];
 8 int n,d,m;
 9 
10 int gcd(int a,int b)
11 {
12     return (a==0)?b:gcd(b%a,a);
13 }
14 
15 int get(int x1,int y1,int x2,int y2,int t)
16 {
17     int dx=x2-x1;
18     int dy=y2-y1;
19     if(dx==0&&dy==0) return c[x2][y2][t];
20     int sum=0;
21     if(dx==0)
22     {
23         if(y1>y2) swap(y1,y2);
24         for(int i=y1; i<=y2; i++)
25         {
26             sum+=c[x2][i][t];
27         }
28         return sum;
29     }
30     if(dy==0)
31     {
32         if(x1>x2)swap(x1,x2);
33         for(int i=x1; i<=x2; i++)
34         {
35             sum+=c[i][y1][t];
36         }
37         return sum;
38     }
39     int g=gcd(abs(dx),abs(dy));
40     dx=dx/g;
41     dy=dy/g;
42     for(int i=0; i<=g; i++)
43     {
44         sum+=c[x1+i*dx][y1+i*dy][t];
45     }
46     return sum;
47 }
48 
49 int main()
50 {
51     while(scanf("%d%d%d",&n,&d,&m)!=EOF)
52     {
53         int max1=0;
54         memset(dp,0,sizeof(dp));
55         memset(c,0,sizeof(c));
56         if(n==0&&d==0&&m==0) break;
57         for(int i=0; i<m; i++)
58         {
59             int x,y,t1;
60             scanf("%d%d%d",&x,&y,&t1);
61             max1=max(max1,t1);
62             c[x+d][y+d][t1]=1;
63         }
64         n=n+2*d;
65         for(int t=1; t<=max1; t++)
66         {
67             for(int i=0; i<n; i++)
68             {
69                 for(int j=0; j<n; j++)
70                 {
71                     for(int x2=0; x2<n ; x2++)
72                     {
73                         for(int y2=0; y2<n; y2++)
74                         {
75                             if((i-x2)*(i-x2)+(j-y2)*(j-y2)<=d*d)
76                             {
77                                 int ans=get(x2,y2,i,j,t);
78                                 dp[t][i][j]=max(dp[t][i][j],dp[t-1][x2][y2]+ans);
79                             }
80                         }
81                     }
82                 }
83             }
84         }
85         int max2=0;
86         for(int i=0; i<n; i++)
87         {
88             for(int j=0; j<n; j++)
89             {
90                 max2=max(max2,dp[max1][i][j]);
91             }
92         }
93         printf("%d\n",max2);
94     }
95     return 0;
96 }