求数组中两两相加等于20的组合(Python实现)

Posted terry-c

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了求数组中两两相加等于20的组合(Python实现)相关的知识,希望对你有一定的参考价值。

题目

求数组中两两相加等于20的组合。
例:给定一个数组[1, 7, 17, 2, 6, 3, 14],这个数组中满足条件的有两对:17+3=20, 6+14=20。

解析

分为两个步骤:

  1. 先采用堆排序或快速排序对数组进行排序,时间复杂度为O(nlogn)。
  2. 然后对排序的数组分别从前到后和从后到前进行遍历, 时间复杂度为O(n)。

假设从前到后遍历的下标为begin,从后到前遍历的下标为end。

  • 当arr[begin] + arr[end] < 20时,满足条件的数一定在[begin+1, end]之间;
  • 当arr[begin] + arr[end] > 20时,满足条件的数一定在[begin, end-1]之间;
  • 当arr[begin] + arr[end] = 20时,找到一组符合条件的数,剩下的组合一定在[begin-1, end-1]之间。

整个算法的时间复杂度为O(nlogn)

Python实现

# -*- coding:utf-8 -*-


def quick_sort(arr, left, right):
    """快速排序"""
    if left >= right:
        return
    low = left
    high = right
    p = arr[left]
    while left < right:
        while left < right and arr[right] >= p:
            right -= 1
        arr[left] = arr[right]
        while left < right and arr[left] <= p:
            left += 1
        arr[right] = arr[left]
    arr[left] = p

    quick_sort(arr, low, left-1)
    quick_sort(arr, left+1, high)


def find_sum(arr, sum):
    """寻找数组中相加等于sum的组合"""
    quick_sort(arr, 0, len(arr) - 1)
    begin, end = 0, len(arr) - 1
    while begin < end:
        if arr[begin] + arr[end] < sum:
            begin += 1
        elif arr[begin] + arr[end] > sum:
            end -= 1
        else:
            print(‘%s %s‘ % (arr[begin], arr[end]))
            begin += 1
            end -= 1


if __name__ == ‘__main__‘:
    arr = [1, 7, 17, 2, 6, 3, 14]
    find_sum(arr, 20)

参考资料

  1. 《Java程序员面试笔试宝典》

以上是关于求数组中两两相加等于20的组合(Python实现)的主要内容,如果未能解决你的问题,请参考以下文章

求一个数组的数字有几种组合方式,6个数相加等于100.用PHP写的程式

java编程:从数组(1,2,4,6,9)中列出所有相加等于20的组合,例如9+9+2=20

算法-求两个有序数组两两相加的值最小的K个数

python 给定数组任意组合等于一个定值的所有解

求数组中,一共有几对 每两个值相加等于8的

为了去小米面试准备的算法题:求递增数组中相加等于10的元素对