数组中和等于K的数对

2021-01-03   96 次阅读


/*
 * 题目:数组中和等于K的数对 
 * 描述:给出一个整数K和一个无序数组A,A的元素为N个互不相同的整数,
 * 		 找出数组A中所有和等于K的数对。例如K = 8,数组A:{-1,6,5,3
 *       ,4,2,9,0,8},所有和等于8的数对包括(-1,9),(0,8),(2,6),(3,5)。
 * Input: 
 *   第1行:用空格隔开的2个数,K N,N为A数组的长度。(2 <= N <= 50000,-10^9 <= K <= 10^9)
 *   第2 - N + 1行:A数组的N个元素。(-10^9 <= Aii <= 10^9)
 * Output:
 *   第1 - M行:每行2个数,要求较小的数在前面,并且这M个数对按照较小的数升序排列。 
 *   如果不存在任何一组解则输出:No Solution。
 */
 
#include <stdio.h>

int a[50001];
int k, n, sum;
// -1	9	7	3	5	4

void quickSort(int left, int right)
{
    if (left < right)
    {
    	
        int i = left;
		int j = right;
		int x = a[left];
        while (i < j)
        {
            while(i < j && a[j] >= x) // 从右向左找第一个小于x的数
                j--;  
            if(i < j) 
                a[i++] = a[j];
            
            while(i < j && a[i] < x) // 从左向右找第一个大于等于x的数
                i++;  
            if(i < j) 
                a[j--] = a[i];
        }
        a[i] = x;
        quick_sort(left, i - 1); // 递归调用 
        quick_sort(i + 1, right);
    }
}

int main() {

	bool key = false;	//用来监测是否有解,无解则输出 No Solution 
		
	scanf("%d %d", &k, &n);
	
	for (int i = 0; i < n; i++) {
		scanf("%d", &a[i]);
	}
	quickSort(0, n - 1);
	for (int i = 0; i < n; i++) {
		printf("%d ", a[i]);
	}
	printf("\n");
	
	int i = 0;
	int j = n - 1;
	while (i < j) {
		sum = a[i] + a[j];
		if(sum == k) {
			key = true; 
			printf("%d %d\n", a[i], a[j]);
			++i;
			--j;
		} else if (sum > k) {
			--j;
		} else {
			++i;
		}
	}
	
	if (key == false) {
		printf("No Solution");
	}
	
	return 0;
} 
 

Q.E.D.

知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议

If you don't come, I will snow.