Skip to content

Latest commit

 

History

History
151 lines (120 loc) · 4.72 KB

Solution.md

File metadata and controls

151 lines (120 loc) · 4.72 KB

P1005 矩阵取数游戏

题目传送门

题目描述

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

每次取数时须从每行各取走一个元素,共 n 个。经过 m 次后取完矩阵内所有元素;
每次取走的各个元素只能是该元素所在行的行首或行尾;
每次取数都有一个得分值,为每行取数的得分之和,每行取数的得分 = 被取走的元素值×2的i次方,其中 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

说明/提示

NOIP 2007 提高第三题。

数据范围:

60% 的数据满足:1 ≤ n,m ≤ 30,答案不超过 10的16次方
100% 的数据满足:1 ≤ n,m ≤ 80,0 <= a{i,j} <= 1000

思路

DP

分别计算出每一行取数m次的最大值,每一行结果之和即最终答案
对每一行:
f[i][j][0/1]表示第i次取第j个数,并且是从行首(0)或行尾(1)取的数

转移方程:

f[i][j][0] = row[j]*pow(2, i) + max(f[i - 1][j - 1][0], f[i - 1][m - i + j + 1][1])
f[i][j][1] = row[j]*pow(2, i) + max(f[i - 1][j + 1][1], f[i - 1][i + j - m - 1][0])
意思是由上一次取数转移而来,上一次取数可以是行首,也可以是行尾,选择其中大的
若第i次和第i - 1次在相同方向取数(都从行首,或都从行尾)
f[i][j][0] 从 f[i - 1][j - 1][0]转移过来
f[i][j][1] 从 f[i - 1][j + 1][1]转移过来
若第i次和上一次在不同方向取数:
由于第i次取数开始之前,已经有 i - 1 个数被取走,
且第i次要取第j个数, 若第i次取行首(即第j个数是行首),那么可以计算出当前行尾为 m - i + j, 那么若上一次取行尾,就是取的当前行尾+1,即 m - i + j + 1
若第i次取行尾(即第j个数是行尾),那么可以计算出当前行首为 i - m + j, 那么若上一次取行首,就是取的当前行首-1, 即 i - m + j - 1
需要注意i,j枚举的范围以及特殊情况(见代码注释)

初始化:

f[1][1][0] = row[1]*2
f[1][m][1] = row[m]*2
另外,对于每次取数,需要单独考虑行首为第1个数,行尾为最后一个数的情况,这两种情况下,上一次取数不能取行首,行尾(见代码注释)

Code

#include <stdio.h>
#include <stdlib.h>

#define maxn 80
#define maxm 80
int m;

//typedef long long ll;
typedef __uint128_t ll;
ll f[maxm + 1][maxm + 1][2];
int row[maxm + 1];

static ll pow(int i) {
    return (ll)1 << i;
}

static ll max(ll a, ll b) {
    return a > b ? a : b;
}

static ll dp() {
    f[1][1][0] = row[1] * 2;
    f[1][m][1] = row[m] * 2;
    for (int i = 2; i <= m; i++) {
        f[i][1][0] = row[1]*pow(i) + f[i - 1][m - i + 2][1]; //行首为1,上一次取数只能从行尾取
        for (int j = 2; j < i; j++)
            f[i][j][0] = row[j]*pow(i) + max(f[i - 1][j - 1][0], f[i - 1][m - i + j + 1][1]);
        f[i][i][0] = row[i]*pow(i) + f[i - 1][i - 1][0]; //行首为i,说明之前每次取数均从行首取

        f[i][m][1] = row[m]*pow(i) + f[i - 1][i - 1][0]; //行尾为m, 上一次取数只能从行首取
        for (int j = m - 1; j > m - i + 1; j--)
            f[i][j][1] = row[j]*pow(i) + max(f[i - 1][j + 1][1], f[i - 1][i + j - m - 1][0]);
        f[i][m - i + 1][1] = row[m - i + 1]*pow(i) + f[i - 1][m - i + 2][1]; //行尾为m - i + 1,说明之前每次均从行尾取
    }
    ll res = 0;
    for (int i = 1; i <= m; i++) {
        res = max(res, f[m][i][0]);
        res = max(res, f[m][i][1]);
    }
//    printf("%lld\n", res);
    return res;
}

static char *itoa(ll i) {
    size_t len = 0;
    ll j = i;
    while (j) {
        j /= 10;
        len++;
    }
    char *str = malloc((len + 1) * sizeof(char));
    str[len] = '\0';
    while (i) {
        str[--len] = i % 10 + '0';
        i /= 10;
    }
    return str;
}

int main() {
    int n;
    scanf("%d %d", &n, &m);
    ll res = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 1; j <= m; j++)
            scanf("%d", row + j);
        res += dp();
    }
    if (res) {
        char *strres = itoa(res);
        printf("%s\n", strres);
        free(strres);
    }
    else printf("0");
    exit(0);
}