题目描述
一个房间里有 n
个 空闲 座位和 n
名 站着的 学生,房间用一个数轴表示。给你一个长度为 n
的数组 seats
,其中 seats[i]
是第 i
个座位的位置。同时给你一个长度为 n
的数组 students
,其中 students[j]
是第 j
位学生的位置。
你可以执行以下操作任意次:
- 增加或者减少第
i
位学生的位置,每次变化量为 1
(也就是将第 i
位学生从位置 x
移动到 x + 1
或者 x - 1
)
请你返回使所有学生都有座位坐的 最少移动次数 ,并确保没有两位学生的座位相同。
请注意,初始时有可能有多个座位或者多位学生在 同一 位置。
示例 1:
输入:seats = [3,1,5], students = [2,7,4]
输出:4
解释:学生移动方式如下:
- 第一位学生从位置 2 移动到位置 1 ,移动 1 次。
- 第二位学生从位置 7 移动到位置 5 ,移动 2 次。
- 第三位学生从位置 4 移动到位置 3 ,移动 1 次。
总共 1 + 2 + 1 = 4 次移动。
示例 2:
输入:seats = [4,1,5,9], students = [1,3,2,6]
输出:7
解释:学生移动方式如下:
- 第一位学生不移动。
- 第二位学生从位置 3 移动到位置 4 ,移动 1 次。
- 第三位学生从位置 2 移动到位置 5 ,移动 3 次。
- 第四位学生从位置 6 移动到位置 9 ,移动 3 次。
总共 0 + 1 + 3 + 3 = 7 次移动。
示例 3:
输入:seats = [2,2,6,6], students = [1,3,2,6]
输出:4
解释:学生移动方式如下:
- 第一位学生从位置 1 移动到位置 2 ,移动 1 次。
- 第二位学生从位置 3 移动到位置 6 ,移动 3 次。
- 第三位学生不移动。
- 第四位学生不移动。
总共 1 + 3 + 0 + 0 = 4 次移动。
提示:
n == seats.length == students.length
1 <= n <= 100
1 <= seats[i], students[j] <= 100
解法
方法一:排序
将两个数组分别排序,然后遍历两个数组,计算每个学生的座位与其实际座位的距离,将所有距离相加即为答案。
时间复杂度 $O(n \times \log n)$,空间复杂度 $O(\log n)$。其中 $n$ 为数组 seats
和 students
的长度。
| class Solution:
def minMovesToSeat(self, seats: List[int], students: List[int]) -> int:
seats.sort()
students.sort()
return sum(abs(a - b) for a, b in zip(seats, students))
|
| class Solution {
public int minMovesToSeat(int[] seats, int[] students) {
Arrays.sort(seats);
Arrays.sort(students);
int ans = 0;
for (int i = 0; i < seats.length; ++i) {
ans += Math.abs(seats[i] - students[i]);
}
return ans;
}
}
|
1
2
3
4
5
6
7
8
9
10
11
12 | class Solution {
public:
int minMovesToSeat(vector<int>& seats, vector<int>& students) {
sort(seats.begin(), seats.end());
sort(students.begin(), students.end());
int ans = 0;
for (int i = 0; i < seats.size(); ++i) {
ans += abs(seats[i] - students[i]);
}
return ans;
}
};
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 | func minMovesToSeat(seats []int, students []int) (ans int) {
sort.Ints(seats)
sort.Ints(students)
for i, a := range seats {
b := students[i]
ans += abs(a - b)
}
return
}
func abs(x int) int {
if x < 0 {
return -x
}
return x
}
|
| function minMovesToSeat(seats: number[], students: number[]): number {
seats.sort((a, b) => a - b);
students.sort((a, b) => a - b);
return seats.reduce((acc, seat, i) => acc + Math.abs(seat - students[i]), 0);
}
|
1
2
3
4
5
6
7
8
9
10
11
12 | impl Solution {
pub fn min_moves_to_seat(mut seats: Vec<i32>, mut students: Vec<i32>) -> i32 {
seats.sort();
students.sort();
let n = seats.len();
let mut ans = 0;
for i in 0..n {
ans += (seats[i] - students[i]).abs();
}
ans
}
}
|
1
2
3
4
5
6
7
8
9
10
11
12
13 | int cmp(const void* a, const void* b) {
return *(int*) a - *(int*) b;
}
int minMovesToSeat(int* seats, int seatsSize, int* students, int studentsSize) {
qsort(seats, seatsSize, sizeof(int), cmp);
qsort(students, studentsSize, sizeof(int), cmp);
int ans = 0;
for (int i = 0; i < seatsSize; i++) {
ans += abs(seats[i] - students[i]);
}
return ans;
}
|