[NOIP2007 提高组] 矩阵取数游戏
题目描述
帅帅经常跟同学玩一个矩阵取数游戏:对于一个给定的 n×mn \times mn×m 的矩阵,矩阵中的每个元素 ai,ja_{i,j}ai,j 均为非负整数。游戏规则如下:
- 每次取数时须从每行各取走一个元素,共 nnn 个。经过 mmm 次后取完矩阵内所有元素;
- 每次取走的各个元素只能是该元素所在行的行首或行尾;
- 每次取数都有一个得分值,为每行取数的得分之和,每行取数的得分 = 被取走的元素值 ×2i\times 2^i×2i,其中 iii 表示第 iii 次取数(从 111 开始编号);
- 游戏结束总得分为 mmm 次取数得分之和。
帅帅想请你帮忙写一个程序,对于任意矩阵,可以求出取数后的最大得分。
输入格式
输入文件包括 n+1n+1n+1 行:
第一行为两个用空格隔开的整数 nnn 和 mmm。
第 2∼n+12\sim n+12∼n+1 行为 n×mn \times mn×m 矩阵,其中每行有 mmm 个用单个空格隔开的非负整数。
输出格式
输出文件仅包含 111 行,为一个整数,即输入矩阵取数后的最大得分。
样例 #1
样例输入 #1
2 3
1 2 3
3 4 2
样例输出 #1
82
提示
【数据范围】
对于 60%60\%60% 的数据,满足 1≤n,m≤301\le n,m\le 301≤n,m≤30,答案不超过 101610^{16}1016。
对于 100%100\%100% 的数据,满足 1≤n,m≤801\le n,m\le 801≤n,m≤80,0≤ai,j≤10000\le a_{i,j}\le10000≤ai,j≤1000。
【题目来源】
NOIP 2007 提高第三题。
思路
本题用区间dp解决
考虑到每行数字互不相干,可以把每行依次解决后相加
每行用区间dp,由于本题特殊性,可以从大区间推到小区间
在区间 [i,j][i,j][i,j]时,从区间 [i−1,j][i-1,j][i−1,j] 和 [i,j+1][i,j+1][i,j+1] 来
此时选中的数字是第 m+i−j−1m+i-j-1m+i−j−1 个(自行推导)
于是状态转移方程便是
dp[i][j]=max(dp[i−1][j]+2m+i−j+1∗a[i−1],dp[i][j+1]+2m+i−j+1∗a[j+1])dp[i][j]=max(dp[i-1][j]+2^{m+i-j+1}*a[i-1],dp[i][j+1]+2^{m+i-j+1}*a[j+1])dp[i][j]=max(dp[i−1][j]+2m+i−j+1∗a[i−1],dp[i][j+1]+2m+i−j+1∗a[j+1])
注意本题需用高精
我用__int128水过了
还有pow的精度问题,不要用pow,自己预处理2的几次方
代码
#include <iostream>
#include <cstring>
#include <cmath>
using namespace std;
int n,m;
__int128 a[100];
__int128 dp[100][100],ans2,base[200];
inline __int128 read(){
__int128 x = 0, f = 1;
char ch = getchar();
while(ch < '0' || ch > '9'){
if(ch == '-')
f = -1;
ch = getchar();
}
while(ch >= '0' && ch <= '9'){
x = x * 10 + ch - '0';
ch = getchar();
}
return x * f;
}
inline void print(__int128 x){
if(x < 0){
putchar('-');
x = -x;
}
if(x > 9)
print(x / 10);
putchar(x % 10 + '0');
}
void init() {
base[0] = 1;
for (int i = 1; i <= m + 2; i++)
base[i] = base[i - 1] * 2;
}
int main()
{
cin>>n>>m;
init();
while(n--){
memset(dp,0,sizeof(dp));
for(int i=1;i<=m;i++)
a[i]=read();
for(int i=1;i<=m;i++)
for(int j=m;j>=1;j--){
dp[i][j]=max(dp[i][j],dp[i-1][j]+base[m-j+i-1]*a[i-1]);
dp[i][j]=max(dp[i][j],dp[i][j+1]+base[m-j+i-1]*a[j+1]);
}
__int128 ans=0;
for(int i=1;i<=m;i++)
ans=max(ans,dp[i][i]+base[m]*a[i]);
ans2+=ans;
}
print(ans2);
return 0;
}