首页 > 代码库 > POJ 2948 DP

POJ 2948 DP


一个row*col的矩阵,每个格子内有两种矿yeyenum和bloggium,并且知道它们在每个格子内的数量是多少。最北边有bloggium的收集站,最西边有 yeyenum 的收集站。现在要在这些格子上面安装向北或者向西的传送带(每个格子自能装一种)。问最多能采到多少矿。


DP,状态转移方程为

dp[i][j]=Max(dp[i][j-1]+suma[i][j],dp[i-1][j]+sumb[i][j]);

其中sumb[i][j]是第i行1到j列的需要到西边的矿石的价值,suma[i][j]是第j列1到i行需要到最北边的矿石的价值


#include "stdio.h"
#include "string.h"
int dp[510][510],suma[510][510],sumb[510][510],a[510][510],b[510][510];
int Max(int a,int b )
{
    if (a<b) return b;else return a;
}
int main()
{
    int n,m,i,j;

    while (scanf("%d%d",&n,&m)!=EOF)
    {
        if (n+m==0)break;
        for (i=1;i<=n;i++)
            for (j=1;j<=m;j++)
            scanf("%d",&b[i][j]);

        for (i=1;i<=n;i++)
            for (j=1;j<=m;j++)
            scanf("%d",&a[i][j]);



        memset(suma,0,sizeof(suma));
        memset(sumb,0,sizeof(sumb));

        for (i=1;i<=n;i++)
            for (j=1;j<=m;j++)
            suma[i][j]=suma[i-1][j]+a[i][j];
        for (j=1;j<=m;j++)
            for (i=1;i<=n;i++)
            sumb[i][j]=sumb[i][j-1]+b[i][j];

        memset(dp,0,sizeof(dp));

        for (i=1;i<=n;i++)
            for (j=1;j<=m;j++)
            {
                dp[i][j]=Max(dp[i][j-1]+suma[i][j],dp[i-1][j]+sumb[i][j]);


            }
        printf("%d\n",dp[n][m]);

    }
    return 0;
}