三数之和

输入n，求所有符合x^2+y^2+z^2 = n的 x, y, z 的方案数。(x, y, z为非负整数)

 注意事项
n <= 1000000
x,y,z满足(x<=y<=z)，只要选择出来的三个数相同就算同一种方案。

样例
给出 n = 0，返回 1。

解释：
当 x = 0 , y = 0, z = 0时等式成立。

给出 n = 1，返回 1。

解释：
当其中一个为1，剩下两个为0，一共有1种方案。

#ifndef C57_H
#define C57_H
#include<iostream>
#include<vector>
#include<algorithm>
#include<set>
using namespace std;
class Solution {
public:
    /**
    * @param numbers: Give an array numbers of n integer
    * @return: Find all unique triplets in the array which gives the sum of zero.
    */
    vector<vector<int>> threeSum(vector<int> &numbers) {
        // write your code here
        vector<vector<int>> res;
        set<vector<int>> set;
        sort(numbers.begin(), numbers.end());
        for (int i = 0; i < numbers.size() - 2; ++i)
        {
            for (int j = numbers.size() - 1; j>i + 1; --j)
            {
                for (int k = i + 1; k < j; ++k)
                {
                    //存入set中防止重复
                    if (numbers[i] + numbers[k] + numbers[j] == 0)
                        set.insert({ numbers[i], numbers[k], numbers[j] });
                }
            }
        }
        for (auto it = set.begin(); it != set.end(); ++it)
            res.push_back(*it);
        return res;
    }
};
#endif