lintcode1753. 写作业

n个人,他们每个人需要独立做m份作业。
第i份作业需要花费cost[i]的时间。由于每个人的空闲时间不同,第i个人有val[i]的时间,这代表他做作业的总时间不会超过val[i]。每个人都按照顺序,从1号作业开始,然后做2号,3号… 现在,你需要计算出他们最多花了多少的时间。

样例
样例 1:

给定`cost=[1,2,3,5]`,`val=[6,10,4]`,返回`15`。
输入:
[1,2,3,5]
[6,10,4]
输出
15

解释:
第一个人可以完成1号作业,2号作业,3号作业,1+2+3<=6。
第二个人可以完成1号作业,2号作业,3号作业,无法完成4号作业,1+2+3<=101+2+3+5>10。
第三个人可以完成1号作业,2号作业,无法完成3号作业,1+2<=41+2+3>41+2+3+1+2+3+1+2=15,返回15。
样例 2:

给定 `cost=[3,7,3,2,5]`,`val=[10,20,12,8,17,25]`,返回`78`.
输入:
[3,7,3,2,5]
[10,20,12,8,17,25]
输出:
78

解释:
第一个人可以完成1号作业,2号作业, 3 + 7<=10.
第二个人可以完成1号作业,2号作业,3号作业,4号作业和5号作业, 3+7+3+2+5<=20
第三个人可以完成1号作业,2号作业,无法完成三号作业,  3+7<=12,3+7+3>12.
第四个人可以完成1号作业,无法完成2号作业 , 3<=8,7+3>8.
第五个人可以完成1号作业,2号作业,3号作业,4号作业,无法完成5号作业,3+7+3+2<=17,3+7+3+2+5>17.
第六个人可以完成1号作业,2号作业,3号作业,4号作业和5号作业, 3+7+3+2+5<=25
3+7+3+7+3+2+5+3+7+3+3+7+3+2+3+7+3+2+5=78, 返回 78.
注意事项
1<=n<=100000
1<=m<=100000
1<=val[i]<=100000
1<=cost[i]<=100000
class Solution {
     
public:
    /**
     * @param cost: the cost
     * @param val: the val
     * @return: the all cost
     */
    long long doingHomework(vector<int> &cost, vector<int> &val) {
     
        // Write your code here.
        int clen=cost.size();
        int vlen=val.size();
        vector<int>sums(clen,0);
        int sum=0;
        for (int i = 0; i < clen; i++) {
     
            /* code */
            sum+=cost[i];
            sums[i]=sum;
        }
        long long res=0;
        for (auto v : val) {
     
            /* code */
            res+=search(sums,v);
        }
        return res;
    }
    long long search(vector<int> sum,int val)
    {
     
        if(val>=sum[sum.size()-1]) return sum[sum.size()-1];
        else if(val<sum[0]) return 0;
        else
        {
     
            int left=0;
            int right=sum.size()-1;
            while(left<=right)
            {
     
                int mid=left+(right-left)/2;
                if(sum[mid]>val) right=mid-1;
                else if(sum[mid]<=val&&sum[mid+1]>val) return sum[mid];
                else if(sum[mid]<val&&sum[mid+1]<=val) left=mid+1;
            }
        }
    }
};

你可能感兴趣的:(lintcode)