当前位置：首页 > 文章列表 > 文章 > python教程 > Python嵌套循环优化技巧分享

Python嵌套循环优化技巧分享

2025-08-06 11:48:29 0浏览收藏

从现在开始，努力学习吧！本文《Python高效嵌套循环优化技巧》主要讲解了等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

优化Python嵌套循环：计算团队获胜次数

本文旨在优化一个计算团队获胜次数的算法，该算法基于比较两个团队成员的技能值。原始算法的时间复杂度为O(n^2)，通过将问题转化为查找数组中和大于0的数对问题，并结合排序和二分查找，可以将时间复杂度降低到O(n log n)。本文将详细介绍优化过程，并提供Python代码示例。

问题描述

给定两个长度为N的数组 group1 和 group2，分别代表两个团队成员的技能值。需要计算 group1 赢得“回合”的次数。当且仅当 group1[i] + group1[j] > group2[i] + group2[j] 时，group1 赢得一回合，其中 0 <= i < j < N。

原始算法及其局限性

最初的解决方案使用嵌套循环，时间复杂度为O(n^2)：

def countWins_original(group1, group2):
    n = len(group1)
    wins = 0
    for i in range(n):
        for j in range(i+1, n):
            if group1[i] + group1[j] > group2[i] + group2[j]:
                wins += 1
    return wins

虽然简单直观，但当N很大时，这种算法的效率会显著下降。

优化思路

关键在于将比较 group1[i] + group1[j] > group2[i] + group2[j] 转化为 (group1[i] - group2[i]) + (group1[j] - group2[j]) > 0。我们可以创建一个新的数组 differences，其中 differences[i] = group1[i] - group2[i]。现在问题简化为：在 differences 数组中，找到有多少对 (differences[i], differences[j]) 的和大于0。

这个问题可以通过以下步骤解决：

计算差值数组： 创建一个新数组 differences，存储 group1 和 group2 对应元素的差值。
排序： 对 differences 数组进行排序。
计数： 遍历排序后的 differences 数组，对于每个元素 differences[i]，找到有多少个 differences[j] 满足 differences[i] + differences[j] > 0，即 differences[j] > -differences[i]。可以使用二分查找来高效地找到这个数量。

优化后的算法

import bisect

def countWins(group1, group2):
    wins = 0
    differences = [x-y for x,y in zip(group1,group2)]
    differences.sort()
    a = differences
    n = len(differences)
    # Loop to iterate through the array
    for i in range(n): 

        # Ignore if the value is negative
        if (a[i] <= 0):
            continue

        # Finding the index using lower_bound
        j = bisect.bisect_left(a, -a[i] + 1);

        # Finding the number of pairs between
        # two indices i and j
        wins += i - j;
    return wins

def main():
    arr1 = [1,3,4,6]
    arr2 = [0,1,4,7]
    print(countWins(arr1,arr2))

main()

代码解释

differences = [x-y for x,y in zip(group1,group2)]: 使用列表推导式，简洁地计算了 group1 和 group2 的差值数组。zip 函数将两个数组对应位置的元素打包成元组。
differences.sort(): 对差值数组进行排序。排序是 O(n log n) 的操作。
bisect.bisect_left(a, -a[i] + 1): 对于每个 a[i]，使用 bisect.bisect_left 在排序后的数组中找到第一个大于等于 -a[i] + 1 的元素的索引 j。 bisect_left 函数实现了二分查找，时间复杂度为 O(log n)。
wins += i - j: i - j 表示在索引 j 之前，有多少个元素小于等于 -a[i]。因此，索引 i 之前的元素中，有 i - j 个元素与 a[i] 的和大于0。