题目描述
这里有一个nm的矩阵,请你选出其中k个子矩阵,使得这个k个子矩阵分值之和最大。
注意:选出的k个子矩阵 不能相互重叠。
输入描述:
第一行为n,m,k(1 ≤ n ≤ 100,1 ≤ m ≤ 2,1 ≤ k ≤ 10),
接下来n行描述矩阵每行中的每个元素的分值(每个元素的分值的绝对值不超过32767)。
输出描述:
只有一行为k个子矩阵分值之和最大为多少。
题解
注意到m=1或者2
当m=1时,是普通的最大连续字段和,只不过是k个:
设dp[i][j]表示前j个数中取出i个矩形的最大和
dp方程
dp[i][j]=max(dp[i][j],dp[i-1][k]+sum[j]-sum[k]);
当m=2时,设f[i][j][k]表示第一列选到第i个数,第二列选到第j个数时,总共k个子矩形的答案
转移有4种情况
当这一位什么都不做的时候
dp[i][j][k]=max(dp[i-1][j][k],dp[i][j-1][k])
当仅选取第一列的某段区间时
dp[i][j][k]=max(dp[l][j][k-1]+sum[i][1]-sum[l-1][1])
当仅选取第二列的某段区间时
dp[i][j][k]=max(dp[i][l][k-1]+sum[j][2]-sum[l-1][2])
当i==j时,可以选取两列一起的
dp[i][j][k]=max(f[l][l][k]+sum[i][1]+sum[i][2]-sum[l-1][1]-sum[l-1][2])
最后所有情况取最大
代码
#include<bits/stdc++.h> using namespace std; #define fi first #define se second #define pb push_back #define mp make_pair #define lowbit(x) x&(-x) typedef long long ll; typedef pair<int,int> pii; typedef pair<ll, ll> pll; const int N = 5e5+5; const ll mod = 1e9+7; const int INF = 0x3f3f3f3f; const double eps =1e-9; const double PI=acos(-1.0); const int dir[8][2]={-1,0,1,0,0,-1,0,1,1,1,1,-1,-1,1,-1,-1}; ll qpow(ll x,ll y){ ll ans=1,t=x; while(y>0){ if(y&1)ans*=t,ans%=mod; t*=t,t%=mod; y>>=1; } return ans%mod; } int dp[101][101][11]; void solve(){ int n,m,k;cin>>n>>m>>k; if(m==1){ int dp[11][101],sum[101],x; memset(dp,0,sizeof(dp)); for(int i=1;i<=n;i++)cin>>x,sum[i]=sum[i-1]+x; for(int i=1;i<=k;i++){ for(int j=1;j<=n;j++){ dp[i][j]=dp[i][j-1]; for(int k=j-1;k>=0;k--) dp[i][j]=max(dp[i][j],dp[i-1][k]+sum[j]-sum[k]); } } cout<<dp[k][n]; } else{ int s1[101],s2[101],x,y; for(int i=1,x,y;i<=n;i++) cin>>x>>y,s1[i]=s1[i-1]+x,s2[i]=s2[i-1]+y; int kk=k; for(int k=1;k<=kk;k++){ for(int i=1;i<=n;i++){ for(int j=1;j<=n;j++){ dp[i][j][k]=max(dp[i-1][j][k],dp[i][j-1][k]); for(int l=0;l<i;l++)dp[i][j][k]=max(dp[i][j][k],dp[l][j][k-1]+s1[i]-s1[l]); for(int l=0;l<j;l++)dp[i][j][k]=max(dp[i][j][k],dp[i][l][k-1]+s2[j]-s2[l]); if(i==j)for(int l=0;l<i;l++)dp[i][j][k]=max(dp[i][j][k],dp[l][l][k-1]+s1[i]-s1[l]+s2[j]-s2[l]); } } } cout<<dp[n][n][k]; } } int main(){ ios::sync_with_stdio(0); cin.tie(0);cout.tie(0); //int t;cin>>t; //while(t--)solve(),cout<<'\n'; solve(); return 0; }