设计一个算法,对于一个给定的包含n个整数的集合S和另一个给定的整数X,该算法可以在O(nlogn)时间内确定S中是否存在两个元素,使得它们的和恰为X。 C语言代码实现

#include 
#include 
#include 
//快排 时间复杂度O(nlogn)
void quick_sort(int* arr, int l, int r) {
    if (l >= r) return;
    while (l < r) {
        int x = l, y = r;
        int z = arr[l];
        do {
            while (x <= y && arr[x] < z) x++;
            while (x <= y && arr[y] > z) y--;
            if (x <= y) {
                int temp = arr[x];
                arr[x] = arr[y];
                arr[y] = temp;
                x++, y--;
            }
        } while (x <= y);
        quick_sort(arr, l, y);
        l = x;
    }
}
//判断是否满足条件 O(n)时间复杂度
bool is_ok(int* A, int x, int len) {
	quick_sort(A, 0, len - 1);
	int i = 0, j = len - 1;
	while (i < j) {
		if (A[i] + A[j] == x) return true;
		else if (A[i] + A[j] > x) j--;
		else i++;
	}
	return false;
}

int main() {
	int A[8] = {3, 41, 52, 26, 38, 57, 9, 49};
	int x;
	while (scanf("%d", &x) != EOF) {
		printf("%d\n", is_ok(A, x, 8));
	}

	return 0;
}

理论证明

你可能感兴趣的:(算法导论代码实践)