算法 排列硬币(二分法)

前言

题目链接
你总共有 n 枚硬币，并计划将它们按阶梯状排列。对于一个由 k 行组成的阶梯，其第 i 行必须正好有 i 枚硬币。阶梯的最后一行 可能 是不完整的。

给你一个数字 n ，计算并返回可形成 完整阶梯行 的总行数。

示例 1：

输入：n = 5
输出：2
解释：因为第三行不完整，所以返回 2 。
示例 2：

输入：n = 8
输出：3
解释：因为第四行不完整，所以返回 3 。

提示：

1 <= n <= 231 - 1

---

一、迭代法

//迭代
int arrangeCoins(int n){
	// i 是行号，也是硬币数
	for(int i = 1; i <= n; i++)
	{
		n = n - i;
		if(n <= i)
		{
			return i;
		}
	}

	return 0;
}

---

二、二分法

//二分查找:行号 x 必定小于硬币数量 n
int arrangeCoins(int n){
	int low = 0, high = n;
    long mid, coins;
	while(low <= high)
	{
		mid = (high - low) / 2 + low;
		coins = (mid + 1) * mid / 2; //等差数列
		if(coins == n)
		{
			return mid;
		}
		else if (coins > n)
		{
			high = mid - 1;
		}
		else
		{
			low = mid + 1;
		}
	}

	return high;
}

---

牛顿迭代法不适用于 x^2 + x = 2n 的问题，会导致超时或者计算不准确。

//牛顿迭代法
/*

根据等差数列: (x + 1) * x / 2 = n;
x^2 + x = 2n;
x^2 = 2n - x;
该问题需要求出 x 的值, 等价于 x 是 (2n-x) 的平方根.

类比"求 x 的平方根"问题,假设 i 是 x 的平方根, 可得到: i = x/i;

所以我们的硬币排列问题为:  x = (2n-x)/x.
于是不断迭代靠近 (x + ((2n-x)/x)) / 2 即可求出解.

*/
double squart(double x, double n)
{
	double res = (x + ((2*n-x)/x)) / 2;

	if (fabs(res - x) < 0.00000000000001f)
	{
		return res;
	}
	else
	{
		return squart(res, n);
	}
}

int arrangeCoins(int n){
	
	if( n < 2 ) return n;


	return squart(n, n);
}