[题解]CF1774D Same Count One

思路

首先记所有 1 的数量为 num,那么显然有当 nmodnum0 时无解。那么考虑有解的时候该怎么办。

显然对于每一个 ai 序列中,最终 1 的数量为 numn,记作 t;并记 cnti 表示 ai 序列中 1 的数量。

我们希望最终所有的 cnti 都等于 t,并且希望操作步数最小,我们考虑一个显然的贪心:将 cnti>t 的序列中的 1cntj<t 缺失的 1

这样我们每一次的操作都会使 i=1n|cntit| 减少 2,显然是最优的方案。

注意:如果你在交换的时候,一定需要更新 ai,kaj,k,否则有一个很简单的 Hack。因为你不更新,你的程序会认为 a3,1 在第一次操作后还是 0 可以交换。

Code

#include <bits/stdc++.h>
#define fst first
#define snd second
#define re register

using namespace std;

typedef pair<int,int> pii;
const int N = 1e5 + 10;
int n,m;
int cnt[N];
pii del[N];

struct answer{
    int a,b,pos;
};

inline int read(){
    int r = 0,w = 1;
    char c = getchar();
    while (c < '0' || c > '9'){
        if (c == '-') w = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9'){
        r = (r << 3) + (r << 1) + (c ^ 48);
        c = getchar();
    }
    return r * w;
}

inline void solve(){
    int num = 0;
    n = read();
    m = read();
    bool vis[n + 10][m + 10];
    vector<answer> ans;
    for (re int i = 1;i <= n;i++){
        cnt[i] = 0;
        for (re int j = 1;j <= m;j++){
            int x;
            x = read();
            if (x) vis[i][j] = true;
            else vis[i][j] = false;
            num += x;
            cnt[i] += x;
        }
    }
    if (num % n) return puts("-1"),void();
    num /= n;
    for (re int i = 1;i <= n;i++) del[i] = {cnt[i] - num,i};
    sort(del + 1,del + n + 1);
    for (re int i = 1,j = n;i < j;){
        int p = del[i].snd,q = del[j].snd;
        for (re int k = 1;k <= m && del[i].fst && del[j].fst;k++){
            if (!vis[p][k] && vis[q][k]){
                del[i].fst++;
                del[j].fst--;
                vis[p][k] = true;
                vis[q][k] = false;
                ans.push_back({p,q,k});
            }
        }
        if (!del[i].fst) i++;
        if (!del[j].fst) j--;
    }
    printf("%d\n",ans.size());
    for (auto p:ans) printf("%d %d %d\n",p.a,p.b,p.pos);
}

int main(){
    int T;
    T = read();
    while (T--) solve();
    return 0;
}

作者:WaterSun

出处:https://www.cnblogs.com/BeautifulWish/p/17962528/CF1774D

版权:本作品采用「署名-非商业性使用-相同方式共享 4.0 国际」许可协议进行许可。

posted @   BeautifulWish  阅读(20)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 无需6万激活码!GitHub神秘组织3小时极速复刻Manus,手把手教你使用OpenManus搭建本
· C#/.NET/.NET Core优秀项目和框架2025年2月简报
· Manus爆火,是硬核还是营销?
· 一文读懂知识蒸馏
· 终于写完轮子一部分:tcp代理 了,记录一下
more_horiz
keyboard_arrow_up dark_mode palette
选择主题
点击右上角即可分享
微信分享提示