今天就写下一昨天晚上在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 <= costs.length <= 100
costs.length
为偶数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);
}
}
}