(补题)小塔的饭

题目描述

小塔是一位热爱美食的算法研究员,她正在研究如何为朋友们安排美食派对以获得最大的幸福感。她邀请了N位朋友来品尝M种不同的食物,并记录了每位朋友对每种食物的好感度ai,j 。

现在,小塔想要设计一个美食分配方案,使得总好感度最高,同时保证至少有一种食物被超过一半的朋友品尝过。你能帮助她解决这个问题吗?

给定N个人和M种食物,每个人对每种食物有一个好感度ai,j 。你需要找到一个分配方案,使得:

  • A. 每个人恰好选择一种食物
  • B. 至少有一种食物被超过⌊N/2⌋个人选择
  • C. 所有选择的总好感度最大
输入

输入包含多组测试数据。第一行包含一个整数T,表示测试数据的组数。

对于每组测试数据:

第一行包含两个整数N和M,表示人数和食物种类数

接下来N行,每行包含M个整数ai,1,ai,2,...,ai,M ,表示第i个人对第j种食物的好感度

数据范围:

1≤T≤10

1≤n≤100

1≤m≤1000

1≤ai,j≤109

输出

对于每组测试数据,输出一个整数,表示满足条件的最大总好感度。

样例输入
复制代码
2
5 3
1 2 3
4 5 6
7 8 9
10 11 12
13 14 15
3 2
200 100
400 300
500 600
样例输出
复制代码
45
1200

贪心解法(O(nmlogn)):最小损失替换,直接满足约束

  • 枚举 "热门食物" :同样假设最终被超过半数人选择的食物是 x(遍历所有食物)。
  • 计算 "损失" :如果一个人选 x 而非他原本最喜欢的食物,好感度会减少多少(loss[i] = 原最高好感 - a[i][x])。
  • 最小化损失 :为了让 x 被至少 ⌊N/2⌋+1 人选择,需要选损失最小的 ⌊N/2⌋+1 个人 来 "替换"(选 x),这样总损失最小,总好感度最大。

代码

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int t,n,m;
ll a[109][1009];
int main(){
	scanf("%d",&t);
	while(t--){
		scanf("%d%d",&n,&m);
		ll total=0,like[109]{0},ans=0;
		for(int i=0;i<n;++i){
			for(int j=0;j<m;++j){
				scanf("%lld",&a[i][j]);
				like[i]=max(like[i],a[i][j]);
			}
			total+=like[i];
		}
		ll loss[109]{0};
		for(int i=0;i<m;++i){
			for(int j=0;j<n;++j){
				loss[j]=like[j]-a[j][i];
			}
			sort(loss,loss+n);
		    ll bug=0;
		    for(int j=0;j<n/2+1;++j){
			    bug+=loss[j];
		    }
		    ans=max(ans,total-bug);
		}
		printf("%lld\n",ans);
	}
	return 0;
}
相关推荐
weixin_5134499615 小时前
walk_these_ways项目学习记录第九篇(通过行为多样性 (MoB) 实现地形泛化)--学习算法
学习·算法·机器学习
fish_xk15 小时前
c++内存管理
开发语言·c++·算法
Tisfy15 小时前
LeetCode 3740.三个相等元素之间的最小距离 I:今日先暴力,“明日“再哈希
算法·leetcode·哈希算法·题解·模拟·遍历·暴力
汀、人工智能15 小时前
[特殊字符] 第77课:最长递增子序列
数据结构·算法·数据库架构·图论·bfs·最长递增子序列
网域小星球15 小时前
C语言从0入门(十)|二维数组详解与矩阵实战
c语言·算法·矩阵·二维数组·数组遍历
澈20715 小时前
堆排序:高效构建大顶堆实战
数据结构·算法·排序算法
我真不是小鱼16 小时前
cpp刷题打卡记录27——无重复字符的最长子串 & 找到字符串中所有字母的异位词
数据结构·c++·算法·leetcode
XuecWu316 小时前
原生多模态颠覆Scaling Law?解读语言“参数需求型”与视觉“数据需求型”核心差异
人工智能·深度学习·算法·计算机视觉·语言模型
We་ct16 小时前
LeetCode 69. x 的平方根:两种解法详解
前端·javascript·算法·leetcode·typescript·平方
一直不明飞行16 小时前
C++:string,写法s.find(‘@‘) != s.end()是否有问题
开发语言·c++·算法