东华oj进阶19 排队打水问题

问题:19 排队打水问题

问题描述:

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

输入说明 :

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

输出说明 :

最少的花费时间

输入范例 :

3 2
1 2 3

输出范例 :

7

思路:

贪心算法就是要贪,每次选出花费时间最少的那个人去打水
用两个vector表明正在排队的人和正在打水的人,
每次先将打水中的vector填满
再将正在打水的人全部减去剩余打水时间最少的那个人的时间,并将剩余打水时间为0的人给移出打水ing的vector
直到排队列表为没人为止
最后虽然没人排队了,但是肯还有人正在接水,直接遍历接水列表所需时间,加入总耗时即可

代码实现:

#include<bits/stdc++.h>
using namespace std;

bool cmp(int a,int b)
{
    return a>b;
}

int main()
{
    int n,r;
    int res=0;
    cin>>n>>r;
    vector<int> vec(n);
    vector<int> usingVec;
    for(int i=0;i<n;i++)
    {
        cin>>vec[i];
    }

    sort(vec.begin(),vec.end(),cmp);

    while(vec.size()>0)
    {
        for(int i=0;i<r && vec.size()>0 && usingVec.size()<r;i++)
        {
            usingVec.push_back(vec[vec.size()-1]);
            vec.erase(vec.end()-1,vec.end());
        }

        sort(usingVec.begin(),usingVec.end(),cmp);

        int minTime=usingVec[usingVec.size()-1];

        for(int i=usingVec.size()-1;i>=0;i--)
        {
            usingVec[i]-=minTime;
            res+=minTime;
            if(usingVec[i]<=0)
                usingVec.erase(usingVec.end()-1,usingVec.end());
        }

        res+=vec.size()*minTime;
    }
    //此时已经没有人排队了,依次把耗费时间加上即可
    for(int i=0;i<usingVec.size();i++)
    {
        res+=usingVec[i];
    }

    cout<<res<<endl;
}