三数之和 --- lintcode831

点击打开原题

已知：

给定一个数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;
	}
}