1. 程式人生 > >Wannafly挑戰賽26 B冥土追魂

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;
    }
}