Wannafly挑战赛26 B.冥土追魂 贪心

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

题目描述

有一天 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最终能拿走呱太的数量。

 

题解：

贪心的话，Kuroko一定先把一行全部选，然后再切换到另一行。

这样有一行会只选k % m个， 有 k / m行全选。

枚举哪一行选k % m个即可。

代码：

#include <bits/stdc++.h>
#ifdef LOCAL
#define debug(x) cout<<#x<<" = "<<(x)<<endl;
#else
#define debug(x) 1;
#endif

#define chmax(x,y) x=max(x,y)
#define chmin(x,y) x=min(x,y)
#define lson id<<1,l,mid
#define rson id<<1|1,mid+1,r
#define lowbit(x) x&-x
#define mp make_pair
#define pb push_back
#define fir first
#define sec second
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, int> pii;

const int MOD = 1e9 + 7;
const double PI = acos (-1.);
const double eps = 1e-10;
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3f;
const int MAXN = 5e5 + 5;

int a[1005][1005];
struct nodes {
    ll fir, sec;
    int id;
    bool operator < (const nodes & x) const {
        return fir < x.fir || (fir == x.fir && sec > x.sec);
    }
} sum[4111];

int main() {
#ifdef LOCAL
    freopen ("input.txt", "r", stdin);
#endif
    int n, m, k;
    scanf ("%d %d %d", &n, &m, &k);
    int div = k / m;
    int mod = k % m;
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++) {
            scanf ("%d", &a[i][j]);
            sum[i].fir += a[i][j];
        }
        sort (a[i] + 1, a[i] + 1 + m);
        for (int j = m; j > m - mod; j--) sum[i].sec += a[i][j];
    }
    sort (sum + 1, sum + 1 + n);
    if (mod) {
        ll ans = INFLL;
        for (int i = 1; i <= n; i++) {
            int cnt = 0;
            ll tans = sum[i].sec;
            for (int j = 1; j <= n; j++) {
                if (cnt >= div) break;
                if (j == i) continue;
                cnt++;
                tans += sum[j].fir;
            }
            ans = min(ans, tans);
        }
        printf ("%lld\n", ans);
    } else {
        ll ans = 0;
        for (int i = 1; i <= div; i++) ans += sum[i].fir;
        printf ("%lld\n", ans);
    }
    return 0;
}