leetcode两地调度

1.贪心

设最终N个人去A地的费用为P1A,N个人去B地的费用为P2B,最低总费用为sum=P1A+P2B;

首先假设去B地的N个人也去A地,且费用为P2A,则所有人去A地的费用为sumA=P1A+P2A,则把这N个人送去A地与送去B地的费用之差为P2A-P2B

   即sumA=sum+P2A-P2B,

   即sumA-(P2A-P2B)=sum.

  所有人去A地的费用是固定的,欲使sum最小,就要使P2A-P2B最大,则对P2A-P2B排序,取最大的N个减去即可

代码如下:

class Solution {
public:
    int twoCitySchedCost(vector<vector<int>>& costs) {
      int len=costs.size();
      int sum=0;
      vector<int> res;
      for(int i=0;i<len;i++)
      {
            sum+=costs[i][0];//所有人去A地的费用
            res.push_back(costs[i][0]-costs[i][1]);
      }
      sort(res.begin(),res.end());
      for(int i=len-1;i>=len/2;i--)
        sum-=res[i];
      return sum;
    }
};
发布了191 篇原创文章 · 获赞 3 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/qq_38196982/article/details/104776198