做计数

链接：https://ac.nowcoder.com/acm/contest/3003/E
来源：牛客网

题目描述

这一天，牛牛与 牛魔王相遇了――然而这并不在 牛牛期望之中。
 牛魔王不出意料又给 牛牛一道看似很难的题目：求有多少个不同的正整数三元组 (i,j,k)满足√i+√j=√k，且 i×j≤n。

牛牛并不会做，你能略施援手吗？

当两个三元组 (i_1​,j₁​,k₁​),(i₂​,j₂​,k₂​) 满足 i₁≠i₂​ 或 j₁≠j₂ 或 k₁≠k₂​ 时它们被认为是不同的。

输入描述:

第一行，一个正整数 n。
保证 1≤n≤4×10⁷。

输出描述:

输出一行，一个整数表示答案。

输入

1

输出

1

说明

(1,1,4)

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <string>
#include <math.h>
#include <algorithm>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <sstream>
const int INF=0x3f3f3f3f;
typedef long long LL;
const int mod=1e9+7;
#define Bug cout<<"---------------------"<<endl
const int maxn=1e4+10;
using namespace std;
 
vector<int> vt;

int Solve(int n)//返回n的因子个数 
{
    int res = 1;
    for (int i = 2; i * i <= n; i++) {
        if (n % i == 0) {
            int t = 0;
            while (n % i == 0) {
                n /= i;
                t++;
            }
            res *= t + 1;
        }
    }
    if (n > 1)
        res *= 2;
    return res;
}
 
int main()
{
    #ifdef DEBUG
    freopen("sample.txt","r",stdin);
    #endif
 
    int n;
    scanf("%d",&n);
    for(int i=1;i*i<=n;i++)
        vt.push_back(i*i);
    int ans=0;
    for(int i=0;i<vt.size();i++)
        ans+=Solve(vt[i]);
    printf("%d
",ans);
     
    return 0;
}

-