leetcode 1029 两地调度

 

 

今天就写下一昨天晚上在leetcode上做的题吧，题目是这样的：

公司计划面试 2N 人。第 i 人飞往 A 市的费用为 costs[i][0]，飞往 B 市的费用为 costs[i][1]。

返回将每个人都飞到某座城市的最低费用，要求每个城市都有 N 人抵达。

 

示例：

输入：[[10,20],[30,200],[400,50],[30,20]]
输出：110
解释：
第一个人去 A 市，费用为 10。
第二个人去 A 市，费用为 30。
第三个人去 B 市，费用为 50。
第四个人去 B 市，费用为 20。

最低总费用为 10 + 30 + 50 + 20 = 110，每个城市都有一半的人在面试。

 

提示：

1. 1 <= costs.length <= 100
2. costs.length 为偶数
3. 1 <= costs[i][0], costs[i][1] <= 100

这道思路还是挺简单的，先让所有人都去城市a，然后在计算去b地与去a地之间的差价，将这些差价排好序，如果去b地的费用减去a地的费用为负值，这就说明去b地比去a地便宜，表示这个人应该去b地。由于题目说去a，b城市的人数各一半，所以我们在最后加上前半部分的差价，就得到了最优解。

在这里就没有使用java自带的排序方法了，用了之前写的快排的代码。主要为了复习巩固。

class Solution {
	public static int twoCitySchedCost(int[][] costs) {
		int sum=0;
		int[] temp=new int[costs.length];
		for(int i=0;i<costs.length;i++){
			sum+=costs[i][0];//先让全部人去a地
			temp[i]=costs[i][1]-costs[i][0];//计算去b地与去a地的差价;
		}
		sort(temp, 0, temp.length-1);//对差价从小到大进行排序
		for(int i=0;i<costs.length/2;i++){
			sum+=temp[i];
		}
		return sum;
	}
	public static void sort(int[] a,int low,int hight){
		int start=low;
		int end=hight;
		int key=a[low];
		while(start<end){
			while(start<end&&a[end]>=key){
				end--;
			}
			if(a[end]<=key){
				int temp=a[start];
				a[start]=a[end];
				a[end]=temp;
			}
			while(start<end&&a[start]<=key){
				start++;
			}
			if(a[start]>=key){
				int temp=a[end];
				a[end]=a[start];
				a[start]=temp;
			}
		}
		if(low<start){
			sort(a,low,start-1);
		}
		if(end<hight){
			sort(a,end+1,hight);
		}
	}
}