poj 3034 Whac-a-Mole

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

n*n方阵 特定时间 特定位置有 moles 同一时间锤子可以 打一条直线上的moles

直线最大距离为 d 

根据时间 逐层更新数量

注意超出 d 范围的情况 和 锤子停留在方阵外的情况

代码及其注释：

#include<iostream>

#include<stdio.h>

#include<string.h>

#include<queue>

#include<cmath>

#include<stack>

#include<algorithm>



using namespace std;



int sum[15][42][42];//时间 位置 记录最大值

int appear[15][42][42];//时间 位置 是否出现moles

int d;

int FindSum(int x,int y,int x1,int y1,int t)//在第 t 时间从(x.y)到(x1,y1) 可以多少moles

{

    if((x-x1)*(x-x1)+(y-y1)*(y-y1)>d*d)//如果超出范围 返回-1

    return -1;

    if(y>y1)

    {

        swap(y,y1);

        swap(x,x1);

    }

    int kx=x1-x;int ky=y1-y;

    int s=0;

    if(x==x1)

    {

        for(int j=y;j<=y1;++j)

        {

            if(appear[t][x][j]==1)

            ++s;

        }

        return s;

    }

    if(y==y1)

    {

        for(int i=min(x,x1);i<=max(x,x1);++i)

        {

            if(appear[t][i][y]==1)

            ++s;

        }

        return s;

    }

    if(appear[t][x][y]==1)

    ++s;

    if(appear[t][x1][y1]==1)

    ++s;

    if((abs(kx)==2&&(ky==4||ky==2))||((abs(kx)==2||abs(kx)==4)&&ky==2))

    {

        if(appear[t][(x+x1)/2][(y+y1)/2]==1)

        ++s;

        return s;

    }

    if(abs(kx)==3&&ky==3)

    {

        if((kx==3&&appear[t][x+1][y+1]==1)||(kx==-3&&appear[t][x-1][y+1]==1))

        ++s;

        if((kx==3&&appear[t][x+2][y+2]==1)||(kx==-3&&appear[t][x-2][y+2]==1))

        ++s;

        return s;

    }

    return s;

}

int main()

{



   int n,m;

   while(scanf("%d %d %d",&n,&d,&m)!=EOF)

   {

       if(n==0&&d==0&&m==0)

       break;

       memset(appear,0,sizeof(appear));

       int T=0;

       while(m--)

       {

           int x,y,t;

           scanf("%d %d %d",&x,&y,&t);

           T=max(T,t);

           appear[t][x+d][y+d]=1;

       }

       memset(sum,0,sizeof(sum));

       int ans=0;

       for(int t=1,k=1;t<=T;++t,k=k*10)

       {

           for(int x=0;x<n+2*d;++x)

           {

               for(int y=0;y<n+2*d;++y)

               {

                   for(int i=0;i<n+2*d;++i)

                   {

                       for(int j=0;j<n+2*d;++j)

                       {

                           int q=FindSum(x,y,i,j,t);

                           if(q==-1)//无法到达 不更新

                           continue;

                           sum[t+1][i][j]=max(sum[t+1][i][j],sum[t][x][y]+q);//更新

                           if(t==T)

                           ans=max(ans,sum[t+1][i][j]);//找最大值

                       }

                   }

               }

           }

       }

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

   }

   return 0;

}