LeetCode-1029. Two City Scheduling（两地调度）

最新推荐文章于 2024-02-01 18:44:37 发布

原创最新推荐文章于 2024-02-01 18:44:37 发布 · 397 阅读

1 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

70 篇文章

订阅专栏

本文介绍了一个两地调度问题的解决方案，通过贪心算法确定最经济的人员分配方式，使得总成本最低。详细解释了算法思路，并提供了C语言实现代码，包括归并排序用于数据预处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

两地调度

在这里插入图片描述

Two City Scheduling

在这里插入图片描述

解

贪心法：
假设让2N人全部飞往B市，然后其中N人再从B飞往A。每一个人飞往A市，就会多支付cost[i][0]-cost[i][1]（可正可负），因此最优的方案就是让这个插值最小的N人飞往A市。
C语言没有sort，因此随便写了个归并。

int twoCitySchedCost(int** costs, int costsSize, int* costsColSize)
{
    MergeSort(costs, 0, costsSize - 1);

    int i, total = 0;
    int n = costsSize / 2;

    for (i = 0; i < n; i++){
        total += costs[i][0] + costs[i + n][1];
    }

    *costsColSize = 2;

    return total;
}

void Merge(int** a, int left, int mid, int right)
{
    int len = right - left + 1;//数组长度
    int **temp = (int **)malloc(len * sizeof(int *));
    int k = 0;
    int i = left;//前一数组起始
    int j = mid + 1;//后一数组起始

    for (k = 0; k < len; k++){
        temp[k] = (int *)malloc(2 * sizeof(int));
    }
    k = 0;
    while (i <= mid && j <= right){
        if(a[i][0]-a[i][1] < a[j][0]-a[j][1]){
            temp[k][0] = a[i][0];
            temp[k][1] = a[i][1];
            k++;
            i++;
        }else{
            temp[k][0] = a[j][0];
            temp[k][1] = a[j][1];
            k++;
            j++;
        }
    }
    while(i <= mid){
        temp[k][0] = a[i][0];
        temp[k][1] = a[i][1];
        k++;
        i++;
    }
    while(j <= right){
        temp[k][0] = a[j][0];
        temp[k][1] = a[j][1];
        k++;
        j++;
    }
    for (k = 0; k < len; k++){
        a[left][0] = temp[k][0];
        a[left][1] = temp[k][1];
        left++;
    }

}

void MergeSort(int** a, int left, int right)
{
    if (left == right){
        return;
    }
    int mid = (left + right) / 2;
    MergeSort(a, left, mid);
    MergeSort(a, mid + 1, right);
    Merge(a, left, mid, right);
}

结果
在这里插入图片描述