求助：Python实现输入数字生成网格行列元组的函数

要实现这个功能，核心是找到最接近正方形的整数组合，分为两种场景：刚好能整除输入数字的因数对，或者允许行列乘积略大于输入数字的近似正方形组合。以下是具体实现：

场景1：返回刚好整除的最接近正方形因数对 #

这个场景适用于需要行列乘积严格等于输入数字的情况（比如输入35返回(5,7)或(7,5)，输入20返回(4,5)）。

import math

def get_exact_grid(n):
    if n <= 0:
        raise ValueError("输入必须是正整数")
    
    # 从平方根向下遍历，找到第一个能整除n的数
    sqrt_n = math.isqrt(n)  # 用整数平方根避免浮点数精度问题
    for i in range(sqrt_n, 0, -1):
        if n % i == 0:
            # 返回(行数, 列数)，可根据需求调换顺序
            return (i, n // i)
    # 处理质数情况，返回(n,1)或(1,n)
    return (n, 1)

测试示例 #

print(get_exact_grid(35))  # 输出(5, 7)
print(get_exact_grid(20))  # 输出(4, 5)
print(get_exact_grid(17))  # 输出(17, 1)（质数情况）

场景2：返回最接近正方形的近似组合（允许乘积大于输入） #

这个场景适用于优先追求接近正方形的网格（比如输入20返回(5,5)）。

import math

def get_square_approx_grid(n):
    if n <= 0:
        raise ValueError("输入必须是正整数")
    
    sqrt_n = math.ceil(math.sqrt(n))
    # 判断是否用正方形即可覆盖，否则计算最小列数
    if sqrt_n * sqrt_n >= n:
        return (sqrt_n, sqrt_n)
    else:
        cols = math.ceil(n / sqrt_n)
        return (sqrt_n, cols)

测试示例 #

print(get_square_approx_grid(20))  # 输出(5, 5)
print(get_square_approx_grid(35))  # 输出(6, 6)（6*6=36≥35）
print(get_square_approx_grid(21))  # 输出(5, 5)（5*5=25≥21）

为什么之前的模运算实现不准确？ #

大概率是因为你没有从平方根开始向下遍历，而是从1开始找因数，这样找到的第一个因数对可能是(1, n)这种极端狭长的组合，而非最接近正方形的。另外如果没有处理“允许乘积大于输入”的场景，也会漏掉(5,5)这种近似正方形的结果。

内容的提问来源于stack exchange，提问作者Shuvro Basu