三数之和 --- lintcode831

于 2018-03-14 20:26:22 发布 · 283 阅读

文章标签：

#平方数

Lintcode 专栏收录该内容

34 篇文章

订阅专栏

本文介绍了一种算法，用于计算给定整数n时，所有可能的正整数解(a, b, c)的数量，使得a^2 + b^2 + c^2 = n，并确保a ≤ b ≤ c以避免重复计数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

点击打开原题

已知：

给定一个数n，求所有满足 a^2 + b^2 + c^2的abc组合的数目

思路：

题目唯一算得上难点的就是如何保证唯一性，即计算过123，就不计算213这样的组合。方法就是假设a<b<c即可。

public class Solution {
    /**
     * @param n: an integer
     * @return: the number of solutions
     */
   public int threeSum2(int n) {
		int ret = 0;
		for (int i = 0; i * i * 3 <= n; i++) {
			for (int j = i; j * j * 2 <= n - i * i; j++) {
				if (calculate(n - i * i - j * j)) {
					ret++;
				}
			}
		}

		return ret;
	}

	private boolean calculate(int num) {
		int i;
		for (i = 0; i * i < num; i++) {

		}
		return i * i == num;
	}
}