蓝桥杯 排队打水问题

问题描述

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

输入格式

  第一行n，r (n<=500,r<=75)
  第二行为n个人打水所用的时间Ti (Ti<=100)；

输出格式

  最少的花费时间

样例输入

   3 2
   1 2 3

样例输出

  7

思路解析

• 按照时间从小到大的顺序排序后即为打水的顺序
• 讲打水时间储存在数组中，并且用 vector 循环储存 r 次
• 算出每个 vector 中的时间和

一种最佳打水方案是，将N个人按照Ti从小到大的顺序依次分配到M个龙头打水。
例如样例中，Ti从小到大排序为1，2，3，4，5，6，7，将他们依次分配到3个龙,则

  去龙头一打水的为 1，4，7；
  去龙头二打水的为 2， 5；
  去龙头三打水的为 3，6。

第一个龙头打水的人总花费时间 = 13+42+71=18
第二个龙头打水的人总花费时间 = 22+51=9
第三个龙头打水的人总花费时间 = 32+6*1=12

所以总的等待时间 = 18+9+12=39

代码

#include <iostream>
#include <math.h>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
	int n,r,sum=0;
	cin>>n>>r;
	int a[1000];
	for(int i=0;i<n;i++)
		cin>>a[i];
	for(int j=0;j<r;j++)
	{
		int i=0;
		vector<int> b;
		while(j+i<n)
		{
			b.push_back(a[j+i]);
			i+=r;
		}
		int x=b.size();
		for(int i=0;i<b.size();i++)
		{
			sum+=b[i]*x;
			x-=1;
		}
	}
	cout<<sum<<endl;
}

输出结果