链接: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;
}
}