优秀的编程知识分享平台

网站首页 > 技术文章 正文

C语言解决荷兰国旗问题

nanyue 2025-04-24 06:32:32 技术文章 3 ℃

荷兰国旗问题是将一个包含三种元素的数组排序的问题,通常的做法是用三个指针:low、mid、high。初始时,low和mid都指向数组的起始位置,high指向末尾。然后通过遍历数组,根据mid指针所指的元素的值来进行交换和移动指针。

以下是完整的C语言代码示例:

#include <stdio.h>

// 交换两个整数的值
void swap(int *a, int *b) {
    int temp = *a;
    *a = *b;
    *b = temp;
}

// 荷兰国旗算法实现
void dutchFlagSort(int arr[], int n) {
    int low = 0;       // 指向当前0的区域的末尾+1
    int mid = 0;       // 遍历指针
    int high = n - 1;  // 指向当前2的区域的开头-1

    // 当mid指针未超过high时继续处理
    while (mid <= high) {
        switch (arr[mid]) {
            case 0:
                // 如果遇到0,交换到low区域
                swap(&arr[low], &arr[mid]);
                low++;  // low区域向右扩展
                mid++;  // mid指针前进(因为交换过来的值已经被检查过)
                break;
            case 1:
                // 遇到1直接跳过
                mid++;
                break;
            case 2:
                // 遇到2,交换到high区域
                swap(&arr[mid], &arr[high]);
                high--; // high区域向左扩展
                // 注意:这里不需要移动mid指针,因为交换过来的值尚未检查
                break;
        }
    }
}

// 打印数组
void printArray(int arr[], int n) {
    for (int i = 0; i < n; i++) {
        printf("%d ", arr[i]);
    }
    printf("\n");
}

int main() {
    int arr[] = {2, 0, 1, 1, 2, 0, 1, 2};
    int n = sizeof(arr) / sizeof(arr[0]);

    printf("原始数组: ");
    printArray(arr, n);

    dutchFlagSort(arr, n);

    printf("排序结果: ");
    printArray(arr, n);

    return 0;
}

代码执行结果

原始数组: 2 0 1 1 2 0 1 2 
排序结果: 0 0 1 1 1 2 2 2 

详细解释

1. 算法思想

使用三个指针将数组划分为三个区域:

  • low 左侧:已排序的0
  • low 到 mid:已排序的1
  • high 右侧:已排序的2
  • mid 到 high:待处理区域

2. 核心逻辑

  • 遇到0:与low位置的元素交换,low和mid都右移(因为交换到mid位置的元素必定是1)
  • 遇到1:直接跳过,mid右移
  • 遇到2:与high位置的元素交换,high左移(但mid不动,因为交换过来的元素可能是0或1)

3. 时间复杂度

  • O(n):只需一次遍历
  • 空间复杂度O(1):原地排序

4. 关键点

  • mid <= high 是循环终止条件
  • 当处理2时,mid指针不移动(因为交换来的新值需要重新检查)

适用场景

  • 三值排序问题
  • 需要线性时间且原地操作的排序需求
  • 扩展可用于多色排序(需调整指针数量)

这个实现是荷兰国旗问题的经典解决方案,兼顾效率与简洁性。

Tags:

最近发表
标签列表