一个房间里有 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
方法一:排序
将两个数组分别排序,然后遍历两个数组,计算每个学生的座位与其实际座位的距离,将所有距离相加即为答案。
时间复杂度 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;
}
}
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;
}
};
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);
const n = seats.length;
let ans = 0;
for (let i = 0; i < n; i++) {
ans += Math.abs(seats[i] - students[i]);
}
return ans;
}
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
}
}
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;
}