hdu 1078 FatMouse and Cheese 메모리 검색/동적 계획

제목: n*n의 행렬, 쥐 한 마리가 (0,0)점에서 출발하여 매번 수평 또는 세로로만 k보를 넘지 못하고, 머무르는 위치의 치즈의 값은 반드시 앞의 위치보다 높아야 하며, 쥐가 치즈를 얼마나 많이 먹을 수 있는지 구한다.
이 문제는 기억화 검색으로 쉽게 풀 수 있다.치즈의 많고 적음에 따라 순서를 정하고 동적 기획을 할 수도 있다.
기억 검색 코드:
#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 110
using namespace std;

int a[N][N],n,k,d[N][N];

int dfs(int x,int y)
{
    if(!d[x][y])
    {
        int ans=0;
        for(int i=-k;i<=k;i++)
        {
            if(i==0)    continue;
            int tx=x+i,ty=y+i;
            if(tx>=0&&tx<n&&a[tx][y]>a[x][y])
                ans=max(ans,dfs(tx,y));
            if(ty>=0&&ty<n&&a[x][ty]>a[x][y])
                ans=max(ans,dfs(x,ty));
        }
        d[x][y]=ans+a[x][y];
    }
    return d[x][y];
}

int main()
{
    while(~scanf("%d%d",&n,&k)&&!(n==-1&&k==-1))
    {
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&a[i][j]);
        memset(d,0,sizeof(d));
        dfs(0,0);
        cout<<d[0][0]<<endl;
    }
}

동적 계획 코드:
#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 11000
using namespace std;

struct node
{
    int c,num;
}b[N];

bool cmp(node a,node b)
{
    return a.c<b.c;
}

int n,k,d[N],a[N];

int main()
{
    while(~scanf("%d%d",&n,&k)&&!(n==-1&&k==-1))
    {
        for(int i=0;i<n*n;i++)
        {
            scanf("%d",&a[i]);
            d[i]=b[i].c=a[i];
            b[i].num=i;
        }
        sort(b,b+n*n,cmp);
        for(int i=n*n-1;i>=0;i--)
        {
            for(int j=-k;j<=k;j++)
            {
                if(j==0) continue;
                int tp=b[i].num;
                int x=tp/n,y=tp%n;
                int t1=(x+j)*n+y,t2=x*n+y+j;
                if(x+j>=0&&x+j<n&&a[t1]>a[tp])
                    d[tp]=max(d[tp],d[t1]+a[tp]);
                if(y+j>=0&&y+j<n&&a[t2]>a[tp])
                    d[tp]=max(d[tp],d[t2]+a[tp]);
            }
        }
        cout<<d[0]<<endl;
    }
}

좋은 웹페이지 즐겨찾기