找和为k的两个元素

今天在做coursera上的练习题时,遇到了一个找和为k的两个元素的问题,按照常规的方法,将数据存入数组中,一次为每个数字遍历数组,找到是否存在和为k的元素,但是这样的话时间复杂度为O(n^2),恰好前两天刚看过快速排序的思想,发现,使用快速排序,可以优化降低时间复杂度。


找和为K的两个元素
描述

在一个长度为n(n < 1000)的整数序列中,判断是否存在某两个元素之和为k。

输入

第一行输入序列的长度n和k,用空格分开。

第二行输入序列中的n个整数,用空格分开。

输出

如果存在某两个元素的和为k,则输出yes,否则输出no。

样例输入

9 10
1 2 3 4 5 6 7 8 9

样例输出

yes

/*
 *  Created on: 2017年8月3日
 *      Author: karo
 *      代码问题:输入n,k,分别表示数据的长度和目标数字,在长度为n的数据中找到是否有两个数相加等于k
 *      解决思路:设置岗哨变量flag。如果k是偶数,flag = k/2+1;如果k是偶数 flag = k/2;
 *      生命一个长度为n+1的数组,用于储存flag和输入的数据,并且令a[0] = flag;
 *      如果k是偶数,遍历数组,找到是否有大于等于2的数字等于flag如果有打印yes,程序结束。
 *      如果没有和k是奇数一样处理。
 *      k是奇数时,将数组分为比flag大的数据和比k小的数据,用比flag大的数,去匹配比flag小的数。
 */
#include
using namespace std;
int quicksort(int *v,int n){
        int key = v[0];
        int low = 0;
        int high = n+1;
        while(low < high)
        {
            while(low < high && v[high] <= key)
            {
                high--;
            }
            v[low] = v[high];
            while(low < high && v[low] >= key)
            {
                low++;
            }
            v[high] = v[low];
        }
        v[low] = key;
        return low;
    return 0;
}

int findk(int *a,int n,int k, int flag){
    int f = quicksort(a,n);
    for(int i=0;ifor(int j=f;j<=n;j++){
            if(a[i]+a[j] == k){
                cout<<"yes"<return 0;
            }
        }
    }
    cout<<"no"<return 0;
}


int main(){
    int n,k,flag;
    int num = 0;
    cin>>n>>k;
    int a[n+1];
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    /*
     * 判断k是奇数还是偶数
     */
    if(k%2 == 1){
        flag = k/2 +1;
//      cout<
    } else{
        flag = k/2;
        for(int i=1;i<=n;i++){
            if(a[i]==flag){
                num++;
            }
        }
        if(num>1){
            cout<<"yes"<return 0;
        }
//      cout<
    }
    a[0] = flag;
    findk(a,n,k,flag);
    return 0;
}

你可能感兴趣的:(入门)