kecode / bigdata Goto Github PK

public static int[] departmentRecruit(int nums[], int[] scores, int[][] preference) {
    // 应聘人数
    int number = scores.length;
    // 分数和排名 i,j i表示排名，越小排名越高，arr[i][0]是原编号 arr[i][1]是分数
    int[][] scoreAndRank = new int[number][2];
    // 记录原有排序
    for (int i = 0; i < number; i++) {
        scoreAndRank[i][0] = i;
        scoreAndRank[i][1] = scores[i];
    }
    // 先算出scoreAndRank
    for (int i = 1; i < number; i++) {
        for (int j = i; j > 0; j--) {
            if (scores[j] > scores[j - 1]) {
                int temp = scores[j];
                scores[j] = scores[j - 1];
                scores[j - 1] = temp;
                // 编号顺序分数交换
                int[] tempArr = scoreAndRank[j];
                scoreAndRank[j] = scoreAndRank[j - 1];
                scoreAndRank[j - 1] = tempArr;
            }
        }
    }
    // 完成招聘计划部门数
    int num1 = 0;
    // 被雇佣人数
    int num2 = 0;
    // 被雇佣了的人 i编号 hired[i] 是否被雇佣 0否1是
    int[] hired = new int[number];
    // 开始计算
    for (int i = 0; i < nums.length; i++) {
        int num = nums[i];
        for (int j = 0; j < number; j++) {
            // 编号
            int code = scoreAndRank[j][0];
            // 人没招满 且 此人没被雇佣
            if (num > 0 && 0 == hired[code]) {
                // 判断是否对此部门有意
                for (int k = 0; k < preference[code].length; k++) {
                    if (i == preference[code][k]) {
                        num2++;
                        hired[code] = 1;
                        num--;
                        break;
                    }
                }
            }
            // 已经招满了
            if(0 == num){
                break;
            }
        }
        if(0 == num){
            num1 ++;
        }
    }
    return new int[]{nums.length - num1,number - num2};
}