洛谷P1005矩阵取数题解--zhengjun

最新推荐文章于 2021-07-19 11:35:58 发布

A_zjzj

最新推荐文章于 2021-07-19 11:35:58 发布

阅读量498

点赞数 1

文章标签： c++ 动态规划动态规划求解算法矩阵

本文链接：https://blog.csdn.net/A_zjzj/article/details/104261206

版权

题目描述

帅帅经常跟同学玩一个矩阵取数游戏：对于一个给定的 $\times m$ 的矩阵，矩阵中的每个元素 $a_{i,j}$
均为非负整数。游戏规则如下：

每次取数时须从每行各取走一个元素，共 $n$ 个。经过 $m$ 次后取完矩阵内所有元素；
每次取走的各个元素只能是该元素所在行的行首或行尾；
每次取数都有一个得分值，为每行取数的得分之和，每行取数的得分 = 被取走的元素值 $×2i \times 2^i$ ，其中 $i$ 表示第 $i$ 次取数（从 $1$ 开始编号）；
游戏结束总得分为 $m$ 次取数得分之和。
帅帅想请你帮忙写一个程序，对于任意矩阵，可以求出取数后的最大得分。

输入格式

输入文件包括 $n + 1$ 行：

第1行为两个用空格隔开的整数 $n$ 和 $m$ 。

第2 $\backsim n+1$ 行为 $\times m$ 矩阵，其中每行有 $m$ 个用单个空格隔开的非负整数。

输出格式

输出文件仅包含 $1$ 行，为一个整数，即输入矩阵取数后的最大得分。

输入输出样例

输入 #1

2 3
1 2 3
3 4 2

输出 #1

说明/提示

$NOIP\ 2007$ 提高第三题

数据范围：

$60\%$ 的数据满足：$1\le n, m \le 30，答案不超过 $10^{16}$

$100\%$ 的数据满足： $1\le n, m \le 8$ ， $\le a_{i,j} \le 1000$

思路

首先，可以很快判断出，这是一道区间动态规划的题，我就干脆打了个递归式的动态规划。
用 $f_{i,j}$ 表示删到头是 $i$ ，尾是 $j$ 时最多的得分
转移公式：
$f_{i,j}=max(f_{i+1,j}+a_i \times 2^{m+i-j-1} ,f_{i,j-1}\times 2^{m+i-j-1})$
最后，这题还会爆 $l o n g l o n g$ 。要么用高精度；要么用 $_\_int128$
所以，我用了 $_\_int128$ ，方便一点。

上代码

#include<bits/stdc++.h>
#define in(x) x=read()
#define MAXN 81
#define bll __int128
using namespace std;
inline void read(register int &x)
{
	x=0;
	register char c=getchar();
	while(c<'0'||c>'9'){c=getchar();}
	while(c>='0'&&c<='9'){x=(x<<3)+(x<<1)+(c^48);c=getchar();}
}
int n,m;
int num[MAXN];
bll ans,p[MAXN],f[MAXN][MAXN];
bll dp(int l,int r)
{
    if(f[l][r]!=-1)return f[l][r];
    if(r-l>=1)f[l][r]=max(num[l]*p[m-r+l]+dp(l+1,r),dp(l,r-1)+num[r]*p[m-r+l]);//因为是倒退，所以是p[m-r+l]
    else f[l][r]=num[l]*p[m-r+l];
    return f[l][r];
}
void print(bll x)
{
    if(!x)return;
    print(x/10);
    putchar(x%10+'0');
}
int main()
{
    read(n);read(m);
    p[0]=1;
    for(int i=1;i<=m;i++)p[i]=p[i-1]*2;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)read(num[j]);
        memset(f,-1,sizeof(f));
        ans+=dp(1,m);
    }
    if(!ans)printf("0")//特判0
    else print(ans);
    return 0;
}