[洛谷]P1005 矩阵取数游戏 (#区间dp)

题目描述

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

1. 每次取数时须从每行各取走一个元素，共nn个。经过mm次后取完矩阵内所有元素；
2. 每次取走的各个元素只能是该元素所在行的行首或行尾；
3. 每次取数都有一个得分值，为每行取数的得分之和，每行取数的得分 = 被取走的元素值\times 2^i×2i，其中ii表示第ii次取数（从11开始编号）；
4. 游戏结束总得分为mm次取数得分之和。

帅帅想请你帮忙写一个程序，对于任意矩阵，可以求出取数后的最大得分。

输入格式

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

第11行为两个用空格隔开的整数nn和mm。

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

输出格式

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

输入输出样例

输入 #1复制

2 3
1 2 3
3 4 2

输出 #1复制

82

说明/提示

NOIP 2007 提高第三题

数据范围：

60%的数据满足：1\le n, m \le 301≤n,m≤30，答案不超过10^{16}1016
100%的数据满足：1\le n, m \le 801≤n,m≤80，0 \le a_{i,j} \le 10000≤ai,j​≤1000

---

思路

有一个n*m大小的矩阵，求n行最大得分和，每一行取数又不会影响到其他行，那么只要确保每一行得分最大就好。

每次取数都在边缘取，那么取数后剩下的元素也一定在某个完整的区间内。于是可以考虑对矩阵每一行进行区间dp。

令dp[i][j]为在矩阵第l行时，区间[i,j]中能获得的最大分数。

这个我们已经比较熟悉了，显然dp[i][j]可以从区间[i-1,j]转移过来，也可以从[i,j+1]转移过来，取个max。可以在第m-j+i-1次取走a[i-1][j]或者a[i][j+1]。

dp[i][j]=max(dp[i-1][j]+a[i-1][j]×2^{m-j+i-1},dp[i][j+1]+a[i][j+1]×2^{m-j+i-1})

最后枚举一下长度为1的区间。答案为

s=max(s,dp[i][i]+a[i][i]×2^{m})

由于本题数据比较毒瘤，即使推对了dp也只能得60。本蒟蒻不会写高精度于是只好偷懒用了__int128。（逃）

注意，__int128不能用cin或cout读，可以自己写一个或者用快读快写读写也行。

我在本题中实现$2$的幂次方是用前缀做的，对于每个$p[i]$代表$2^{i}$。

#include <stdio.h>
#include <iostream>
#include <memory.h>
#include <cmath>
#define r(x) x=read()
#define bll __int128
using namespace std;
bll n,m,k,s,dp[101][101],a[101][101],p[101];
inline int read()
{
	int x(0),w(1);
	char ch=getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch=='-') w=-1;
		ch=getchar();
	}
	while(ch>='0' && ch<='9')
	{
		x=(x<<3)+(x<<1)+ch-'0';
		ch=getchar();
	}
	return x*w;
}
void write(bll x)
{
	if(!x) return;
	if(x) write(x/10);
	putchar(x%10+'0');
}
signed main()
{
	//ios::sync_with_stdio(false);
	//cin.tie(0);
	//cout.tie(0);
	register bll i,j,l;
	r(n);r(m);
	p[0]=1;
	for(i=1;i<=m;i++) p[i]=p[i-1]*2;
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=m;j++)
		{
			r(a[i][j]);
		}
	}
	for(l=1;l<=n;l++)
	{
		for(i=1;i<=m;i++)
		{
			for(j=i;j>=i;j--)//最值是在小区间中，所以要从大到小倒叙 
			{
				dp[i][j]=max(dp[i-1][j]+p[m-j+i-1]*a[l][i-1],dp[i][j+1]+p[m-j+i-1]*a[l][j+1]);
			}
		}
		bll maxn(0);
		for(i=1;i<=m;i++)
		{
			maxn=max(dp[i][i]+a[l][i]*p[m],maxn);//求一整行的最大的分 
		}
		s+=maxn;//加上这一行的分数 
	}
	if(s==0) cout<<0<<endl;
	else write(s);
	cout<<endl;
	return 0;
}