lightoj - 1036 - A Refining Company / dp

最新推荐文章于 2021-05-22 22:33:32 发布

Cosiness

最新推荐文章于 2021-05-22 22:33:32 发布

阅读量365

点赞数

分类专栏：动态规划dp

本文链接：https://blog.csdn.net/cosiness/article/details/50964863

版权

动态规划dp 专栏收录该内容

27 篇文章 0 订阅

订阅专栏

题意：给两张n*m的表格，第一张表格表示Uranium矿的位置，第二张表格表示Radium矿的位置，在表格的最上边是R矿的精炼厂，在表格的最左边是U矿的精炼厂，有两种传送带，一种从右到左，另一种从下到上，表格的一个格子内只能存在一种传送带，如果一个格子内的矿想要被传送带运输，那么这个格子内必须有传送带。传送带必须是直线才能运输，即从右到左的传送带只能运输U矿，从下到上的传送带只能运输R矿，要求修建传送带，使得两种矿的总和最大。

题解：设pre[i][j]为第i行U矿的前缀和，suf[i][j]为第i行R矿的后缀和，dp[i][j]为第i行的第1种传送带最多从第j列修到第1列。

dp[i][j] = max(dp[i][j-1], dp[i-1][j] + pre[i][j] + suf[i][j])

注意是多组数据，所以要初始化。

#include <bits/stdc++.h>
//#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

#define ll long long
#define SZ(x) ((int)(x).size()) 
#define ALL(v) (v).begin(), (v).end()
#define foreach(i, v) for (__typeof((v).begin()) i = (v).begin(); i != (v).end(); ++ i)
#define reveach(i, v) for (__typeof((v).rbegin()) i = (v).rbegin(); i != (v).rend(); ++ i) 
#define REP(i,a,n) for ( int i=a; i<int(n); i++ )
#define FOR(i,a,n) for ( int i=n-1; i>= int(a);i-- )
#define lson rt<<1, L, m
#define rson rt<<1|1, m, R
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
#define mp(x, y) make_pair(x, y)
#define pb(x) push_back(x)
#define fi first
#define se second
#define CLR(a, b) memset(a, b, sizeof(a))
#define Min(a, b) a = min(a, b)
#define Max(a, b) a = max(a, b)
const int maxn = 500 + 7;
int T;
int kase;
int n, m;
int U[maxn][maxn];
int R[maxn][maxn];
int pre[maxn][maxn], suf[maxn][maxn];
int dp[maxn][maxn];
void ini(){
    CLR(U, 0);
    CLR(R, 0);
    CLR(pre, 0);
    CLR(suf, 0);
    CLR(dp, 0);
}
int solve(){
    REP(i, 1, n + 1) REP(j, 1, m + 1) pre[i][j] = pre[i][j-1] + U[i][j];
    REP(i, 1, n + 1) FOR(j, 1, m + 1) suf[i][j] = suf[i][j+1] + R[i][j];
   
    REP(i, 1, n + 1){
        dp[i][0] = dp[i-1][0] + suf[i][1];
        REP(j, 1, m + 1){
            dp[i][j] = max(dp[i][j-1], dp[i-1][j] + pre[i][j] + suf[i][j+1]);
        }
    }
    return dp[n][m];
}
int main(){
#ifdef ac
    freopen("in.txt","r",stdin);
#endif
    //freopen("out.txt","w",stdout);
    scanf("%d", &T);
    while(T--){
        ini();
        scanf("%d%d", &n, &m);
        REP(i, 1, n + 1) REP(j, 1, m + 1) scanf("%d", &U[i][j]);
        REP(i, 1, n + 1) REP(j, 1, m + 1) scanf("%d", &R[i][j]);
        printf("Case %d: %d\n", ++kase, solve());
    }
    return 0;
}