[蓝桥杯][算法提高] ADV-104、148 打水问题 贪心 C++参考代码

ADV-104 VIP试题 打水问题

时间限制: 1Sec 内存限制: 128MB 提交: 554 解决: 271

题目描述
N个人要打水，有M个水龙头，第i个人打水所需时间为Ti，请安排一个合理的方案使得所有人的等待时间之和尽量小。

提示
一种最佳打水方案是，将N个人按照Ti从小到大的顺序依次分配到M个龙头打水。
例如样例中，Ti从小到大排序为1，2，3，4，5，6，7，将他们依次分配到3个龙头，则去龙头一打水的为1，4，7；去龙头二打水的为2,5；去第三个龙头打水的为3,6。
第一个龙头打水的人总等待时间 = 0 + 1 + (1 + 4) = 6
第二个龙头打水的人总等待时间 = 0 + 2 = 2
第三个龙头打水的人总等待时间 = 0 + 3 = 3
所以总的等待时间 = 6 + 2 + 3 = 11

输入
第一行两个正整数N M 接下来一行N个正整数Ti。
N,M< =1000，Ti< =1000

输出
最小的等待时间之和。（不需要输出具体的安排方案）

样例输入
7 3
3 6 1 4 2 5 7

样例输出
11

标签　　
贪心

贪心：

让等候时间短的人先打水

参考代码1（易理解）：

#include<iostream>
#include<algorithm>
#include<vector>
#include<numeric>
using namespace std;

int main(){
    int N,M;
    int Ti;
    cin>>N>>M;
    vector<int>v1;
    vector<int>v2[M];
    for(int i=0;i<N;i++){
        cin>>Ti;
        v1.push_back(Ti);
    }
    sort(v1.begin(),v1.end());
    
    int j=0;
    while(j<N){
        for(int i=0;i<M&&j<N;i++){
            v2[i].push_back(v1[j]);//排好序的数据小到大分别插入M个容器 
            j++;
        }       
    }
    
//  for(int i=0;i<M;i++){
//      for(int j=0;j<v2[i].size();j++){
//          cout<<v2[i][j]<<" ";    
//      }
//      cout<<endl; 
//  } 

    int total=0;//总等待时间 
    int nextwait=0;//前几项的时间累加和 
    for(int i=0;i<M;i++){
        int nextwait=0;
        for(int j=0;j<v2[i].size()-1;j++){
            nextwait+=v2[i][j];
//          cout<<"("<<nextwait<<")";
            total+=nextwait;
        }
//      cout<<"="<<total<<endl;
    }
    
    cout << total;
    return 0;
}

参考代码2:

#include <bits/stdc++.h>
using namespace std;
int ans[1005];
int main(){
    int n,m;
    cin>>n>>m;
    for(int i=0;i<n;i++)
        cin>>ans[i];
    sort(ans,ans+n);
    int s=0;
    for(int i=0;i<n-m;i++){
        ans[i+m]+=ans[i];
        s+=ans[i];
    }
    cout<<s;
}

ADV-148 VIP试题 排队打水问题

时间限制: 1Sec 内存限制: 128MB 提交: 454 解决: 206

题目描述
有n个人排队到r个水龙头去打水，他们装满水桶的时间t1、t2…………tn为整数且各不相等，应如何安排他们的打水顺序才能使他们总共花费的时间最少？

数据规模和约定
其中80%的数据保证n< =10

输入
第一行n，r (n< =500,r< =75)
第二行为n个人打水所用的时间Ti (Ti< =100)；
输出
最少的花费时间
样例输入
3 2
1 2 3
样例输出
7

标签　　标签反馈
贪心

参考代码：

在上一题参考代码1中，去掉-1即可

	int total=0;//总等待时间 
	int nextwait=0;//前几项的时间累加和 
	for(int i=0;i<M;i++){
		int nextwait=0;
		//for(int j=0;j<v2[i].size()-1;j++){
		for(int j=0;j<v2[i].size();j++){
			nextwait+=v2[i][j];
			total+=nextwait;
		}
	}