高精度+dp——[NOIP2007 提高组] 矩阵取数游戏

[NOIP2007 提高组] 矩阵取数游戏

题目描述

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

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

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

输入格式

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

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

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

输出格式

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

样例 #1

样例输入 #1

2 3
1 2 3
3 4 2

样例输出 #1

82

提示

【数据范围】

对于 $60\%$ 的数据，满足 $1\le n,m\le 30$，答案不超过 $10^{16}$。
对于 $100\%$ 的数据，满足 $1\le n,m\le 80$，$0\le a_{i,j}\le 1000$。

【题目来源】

NOIP 2007 提高第三题。

思路

思路来源：
像这样，每次取出一个数，但剩下的数保持完整区间用的就是区间dp。

拓展：能用动态规划解决的问题，需要满足三个条件：最优子结构（就像图片的第一行所说的（管好自己就行）），无后效性（已经求解的子问题，不会再受到后续决策的影响。）和子问题重叠（如果有大量的重叠子问题，我们可以用空间将这些子问题的解存储下来，避免重复求解相同的子问题，从而提升效率）。

状态表示：f[i][j][k]表示在第k行还剩区间[i,j]（这里就用到队列的思想，（区间））
状态计算：f[i][j][k]=max(2*f[i+1][j][k]+2^w[k][i],2*f[i][j-1][k]+2*w[k][j])

代码

其中一行调了半天，就是maxl那

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>

using namespace std;

const int N = 100;

vector<int> f[N][N][N];
int n,m;
vector<int> w[N][N];
vector<int> A,B,C,ans;

vector<int> add(vector<int>& a,vector<int> &b){
    int t=0;
    vector<int>c;
    
    for(int i=0;i<a.size()||i<b.size();i++){
        if(i<a.size())t+=a[i];
        if(i<b.size())t+=b[i];
        c.push_back(t%10);
        t/=10;
    }
    
    if(t){
        c.push_back(t);
    }
    return c;
    
}

vector<int> mul(vector<int> a,int b){
    vector<int>c;
    
    int t=0;
    
    for(int i=0;i<a.size()||t;i++){
        if(i<a.size())t+=a[i]*b;
        c.push_back(t%10);
        t/=10;
    }
    while(c.size()>1&&c.back()==0)c.pop_back();
    return c;
}

vector<int> maxl(vector<int> a,vector<int> b){
    // cout<<a.size()<<' '<<b.size()<<endl;
    if(a.size()>b.size())return a;
    if(a.size()<b.size())return b;
    //这种方式是错的，要从最高位开始比
    // for(int i=0;i<a.size();i++){
    //     if(a[i]>b[i])return a;
    //     else if(a[i]<b[i])return b;
    // }
    
    for(int i=a.size()-1;i>=0;i--){
        if(a[i]>b[i])return a;
        if(a[i]<b[i])return b;
    }
    return a;
}

int main(){
    cin>>n>>m;
    
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            int x;
            cin>>x;
            string a=to_string(x);
            for(int k=a.size()-1;k>=0;k--){
                w[i][j].push_back(a[k]-'0');
            }
        }
    }
    
    for(int i=1;i<=n;i++){
        for(int len=0;len<m;len++){
            for(int l=1;l+len<=m;l++){
                int r=l+len;
                vector<int>a=mul(add(f[l+1][r][i],w[i][l]),2),b=mul(add(f[l][r-1][i],w[i][r]),2);
                f[l][r][i]=maxl(a,b);
                // for(int k=a.size()-1;k>=0;k--)cout<<a[k];
                // cout<<endl;
                // for(int k=b.size()-1;k>=0;k--)cout<<b[k];
                // cout<<endl;
                // // for(int k=f[l+1][r][i].size()-1;k>=0;k--)cout<<f[l+1][r][i][k];
                // // cout<<endl;
                // // cout<<"w=";
                // // for(int k=w[i][l].size()-1;k>=0;k--)cout<<w[i][l][k];
                // // cout<<endl;
                // // vector<int> o=mul(add(f[l+1][r][i],w[i][l]),2);
                // // for(int k=o.size()-1;k>=0;k--)cout<<o[k];
                // // cout<<endl;
                // for(int k=f[l][r][i].size()-1;k>=0;k--)cout<<f[l][r][i][k];
                // cout<<endl;
                // cout<<"______________"<<endl;
            }
        }
    }
    
    for(int i=1;i<=n;i++){
        ans=add(ans,f[1][m][i]);
    }
    
    for(int i=ans.size()-1;i>=0;i--){
        cout<<ans[i];
    }
    
    return 0;
    
}