计蒜客 - 第N个智慧数字

题目链接

一个正整数如果能表示成两个正整数的平方差，则称这个数为一个“智慧数”，比如 1616 就等于 55 的平方减去 33 的平方，所以 1616 就是一个智慧数，从 11 开始的自然数列中，将“智慧数”从小到大编号为 1,2,3,…,n1,2,3,…,n。现输入一个正整数 nn，输出第 nn 个“智慧数”。

输入格式
输入仅包含一个正整数 nn（1\leq n\leq 1001≤n≤100）。

输出格式
输出仅包含一个正整数，表示编号为 nn 的“智慧数”。

样例输入 复制
3
样例输出 复制
7

很久前做的一道题暴力过的，今天偶然看了kbb dalao博客操作
当作复习了一遍stl的set操作
计蒜客上对空间卡的比较紧，多带了一个无关数组都TLE了

#include
#include
using namespace std;
#define ll long long
int main()
{
    ll n,m,a;
    set<long long> st;
    while(cin>>n)
    //满足多组数据在线测试
    //eof结束即可
    {
        for(ll i=1;i<500;i++)
        for(ll j=i+1;j<=500;j++)
        {
            a=j*j-i*i;
            st.insert(a);
        }
        set<long long >::iterator i;
        ll j=1;
        for(i=st.begin();i!=st.end();i++,j++)
        {
            if(j==n) cout<<*i;
        }
        cout<<endl;
    }
    return 0;
}