Wannafly挑战赛26 B冥土追魂

链接：https://www.nowcoder.com/acm/contest/212/B
来源：牛客网
 

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 524288K，其他语言1048576K
64bit IO Format: %lld

题目描述

有一天 Misaka 和 Kuroko 在玩一个关于冥土追魂的游戏....
Misaka和Kuroko在一个 n x m 的棋盘上玩游戏，每个格子上都放着一些呱太。游戏共进行 k 回合，每一回合 Kuroko会选**有呱太**的一行 i，在这之后Misaka会选择一列 j ，并拿走格子 (i, j) 上的所有呱太，Misaka希望自己拿走的呱太尽可能多，而Kuroko不想让Misaka拿走很多呱太，所以她希望拿走的呱太尽可能少。
在一旁围观的恒温死神希望预测结果，请你预测在双方都采取最优策略的情况下，Misaka最终能拿走呱太的数量。

输入描述:

第一行三个数 n, m, k。 
接下来 n 行，每行 m 个数，第 i 行第 j 个数表示棋盘第 i 行第 j 列上的呱太数量 ai,j。

输出描述:

输出共一个数，表示在你的预测下，Misaka最终能拿走呱太的数量。

示例1

输入

复制

3 2 4
5 7
3 2
8 5

输出

复制

17

备注:

1≤ n, m ≤ 1000, 1 ≤ k ≤ n x m, 1 ≤ ai,j ≤ 109

 

解题思路

范围比较小，直接排序然后记录前缀和，然后枚举即可 注意INF要足够大

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAX=1005;
const ll INF=1e18;
int n,m,k;
ll sum[MAX][MAX];
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++) scanf("%lld",&sum[i][j]);
        sort(sum[i]+1,sum[i]+m+1,greater<ll>());
        for(int j=1;j<=m;j++) sum[i][j]+=sum[i][j-1];
    }
    ll ans=0;
    if(k%m==0){
        vector<ll> v;
        for(int i=1;i<=n;i++){
            v.push_back(sum[i][m]);
        }
        sort(v.begin(),v.end());
        for(int i=0;i<k/m;i++) ans+=v[i];
        cout<<ans<<endl;  
    }
    else{
        ans=INF;
        for(int i=1;i<=n;i++){
            ll temp=sum[i][k%m];
            vector<ll> v;
            v.clear();
            for(int j=1;j<=n;j++){
                if(i!=j) v.push_back(sum[j][m]);
            }
            sort(v.begin(),v.end());
            for(int j=0;j<k/m;j++) temp+=v[j];
            if(ans>temp) ans=temp;
        }
        cout<<ans<<endl;
    }
}