题目描述
帅帅经常跟同学玩一个矩阵取数游戏:对于一个给定的 n×m 的矩阵,矩阵中的每个元素 ai,j 均为非负整数。游戏规则如下:
- 每次取数时须从每行各取走一个元素,共 n 个。经过 m 次后取完矩阵内所有元素;
- 每次取走的各个元素只能是该元素所在行的行首或行尾;
- 每次取数都有一个得分值,为每行取数的得分之和,每行取数的得分 = 被取走的元素值 2i,其中 i 表示第 i 次取数(从 1 开始编号);
- 游戏结束总得分为 m 次取数得分之和。
帅帅想请你帮忙写一个程序,对于任意矩阵,可以求出取数后的最大得分。
输入格式
输入文件包括 n+1 行:
第一行为两个用空格隔开的整数 n 和 m。
第 2∼n+1 行为 n×m 矩阵,其中每行有 m 个用单个空格隔开的非负整数。
输出格式
输出文件仅包含 1 行,为一个整数,即输入矩阵取数后的最大得分。
输入输出样例
输入 #1复制
2 3 1 2 3 3 4 2
输出 #1复制
82
说明/提示
【数据范围】
对于 60% 的数据,满足 1≤n,m≤30,答案不超过 1016。
对于 100% 的数据,满足 1≤n,m≤80,0≤ai,j≤1000。
【题目来源】
NOIP 2007 提高第三题。
解析:
我们可以用 dp记录其取数时,先取那个数。
dp[i][j]代表剩下的长度为到 j - i 长度时,先取i-1位置上的数较大,还是取j+1上的数较大。
dp[ i ][j ] = max(dp[i-1][j] + a[i-1]*two[m - (j - i - 1 + 1)],dp[i][j] + a[j+1] *two[m - (j - i - 1 + 1)]);
表示 在 i ~ j 区间 内 ,取 i-1这个位置上的数,还是 j+1这个位置上的数。
这道题 要用高精度:
所以要用__int128
输出函数为
template <typename T > void Out(T x)
{
if(x < 0) putchar('-'),x = -x;
if(x > 9) Out(x/10);
putchar(x%10 +'0');
}
代码:
#include<bits/stdc++.h>
#define rep(i , a , b) for(int i = a ; i <= b ; i ++)
using namespace std;
template <typename T> void Out(T x)
{
if(x < 0) putchar('-') , x = -x;
if(x > 9) Out(x / 10);
putchar(x % 10 + '0');
}
const int M = 1e2 + 10;
__int128 two[M] , dp[M][M] , ans;
int a[M][M];
signed main()
{
two[0] = 1;
rep(i , 1 , M) two[i] = two[i - 1] * 2;//2的阶乘
int n , m;
cin >> n >> m;
for(int i = 1 ; i <= n ; i ++) for(int j = 1 ; j <= m ; j ++) cin >> a[i][j];
for(int i = 1 ; i <= n ; i ++)
{
memset(dp , 0 , sizeof(dp)); // dp的定义为在 长度为 len的情况下 l~ r取最大
for(int len = m ; len >= 1 ; len --)
{
for(int l = 1 ; l + len - 1 <= m ; l ++)
{
int r = l + len - 1;
__int128 c = two[m - (r - l + 1)]; // m - (r-l+1) 这就是 只剩下 m-len的长度
dp[l][r] = max(dp[l - 1][r] + c * a[i][l - 1] , dp[l][r + 1] + c * a[i][r + 1]);//他是倒序的
}
}
__int128 ma = 0;
for(int j = 1 ; j <= m ; j ++)
ma = max(ma , dp[j][j] + two[m] * a[i][j]);
ans += ma;
}
Out(ans) , puts("");
return 0;
}
时间复杂度为:O(m*m*n)