荷兰国旗问题

最新推荐文章于 2022-01-26 12:13:43 发布

原创最新推荐文章于 2022-01-26 12:13:43 发布 · 155 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种解决荷兰国旗问题的算法，该问题要求将数组中的元素根据给定数值进行三向切分，使得小于、等于和大于该数值的元素分别位于数组的左、中、右三个部分。通过使用双指针技术，该算法能够在O(n)的时间复杂度内完成任务。

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

荷兰国旗问题

给定一个数组arr，和一个数num，请把小于num的数放在数组的左边，等于num的数放在数组的中间，大于num的数放在数组的右边。

荷兰国旗问题求解

public class NetherlandsFlag {

	public static int[] partition(int[] arr, int l, int r, int p) {
		int less = l - 1;
		int more = r + 1;
		while (l < more) {
			if (arr[l] < p) {
				swap(arr, ++less, l++);
			} else if (arr[l] > p) {
				swap(arr, --more, l);
			} else {
				l++;
			}
		}
		return new int[] { less + 1, more - 1 };
	}

	// for test
	public static void swap(int[] arr, int i, int j) {
		int tmp = arr[i];
		arr[i] = arr[j];
		arr[j] = tmp;
	}

	// for test
	public static int[] generateArray() {
		int[] arr = new int[10];
		for (int i = 0; i < arr.length; i++) {
			arr[i] = (int) (Math.random() * 3);
		}
		return arr;
	}

	// for test
	public static void printArray(int[] arr) {
		if (arr == null) {
			return;
		}
		for (int i = 0; i < arr.length; i++) {
			System.out.print(arr[i] + " ");
		}
		System.out.println();
	}

	public static void main(String[] args) {
		int[] test = generateArray();

		printArray(test);
		int[] res = partition(test, 0, test.length - 1, 1);
		printArray(test);
		System.out.println(res[0]);
		System.out.println(res[1]);

	}
}