2020快手编程笔试题顾客最小满意度
2020快手笔试题
快手晚饭时间!n个顾客,从1到n标号,一开始每个顾客i在队伍中的位置是i,每个顾客有两个属性ai和bi。
每个顾客的不满意程度等于站在他前面的人与ai的乘积,加上站在他后面的人与bi的乘积。正式来说,假设顾客位于位置j,那么它的不满意程度等于ai * ( j - 1 ) + bi * ( n - j ),
作为食堂经理,本着顾客至上的原则,你需要重新安排每个顾客的位置,使得所有的顾客的不满意度总和最小。
样例
输入
3
8 5
9 8
7 3
输出
[3,1,2]
思路:将公式去括号,加括号,提取出j来,整理得到如下公式:
( ai - bi ) * j -ai +bi*n。这就是每个顾客的不满意度,j是固定的,( -ai + bi*n )也是固定的我们只需要让j小的( ai - bi) 大就可以,所以我们只需要对 (ai - bi)的值排序下(从大到小)就好,注意记录下位置。
代码:
package pers.pta4.t1.t2;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.Scanner;
public class Main {
static Scanner scan = new Scanner(new BufferedReader(new InputStreamReader(System.in)));
static PrintWriter pw = new PrintWriter(System.out, true);
public static void main(String[] args) throws IOException {
int n=scan.nextInt();
int[] a=new int[n];
int[] b=new int[n];
for (int i = 0; i < n; i++) {
a[i]=scan.nextInt();
b[i]=scan.nextInt();
}
int[] ints = new Main().WaitInLine(a, b);
System.out.println(Arrays.toString(ints));
}
public int[] WaitInLine (int[] a, int[] b) {
int len=a.length;
Per[] pers = new Per[len];
for (int i = 0; i < len; i++) {
pers[i]=new Per(i+1,a[i]-b[i]);
}
Arrays.sort(pers);
int[] res=new int[len];
for (int i = 0; i < len ; i++) {
res[i]=pers[i].id;
}
return res;
}
class Per implements Comparable<Per>{
int id;
int value;
public Per(int id, int value) {
this.id = id;
this.value = value;
}
@Override
public int compareTo(Per o) {
return o.value-this.value;
}
}
}