平方根问题

我也是python超时,用网上同样思路的java可以过

import java.lang.Math;
public class Main{
    public static void main(String[] args)
    {
        Scanner sc=new Scanner(System.in);
        int n=Integer.parseInt(sc.next());
        int m=Integer.parseInt(sc.next());
        if(n>m)
        {
            int temp=m;
            m=n;
            n=temp;
        }
        int count=0;
        int sqrtCount=0;
        for(int i=1;i<=n;i++)
        {
            if(Math.sqrt(i)%1==0)
                sqrtCount++;
        }
        for(int i=1;i<=m;i++)
        {
            if(Math.sqrt(i)%1==0)
                count+=sqrtCount;
            else if(nextNum(i)<=n)
                {
                    int base=nextNum(i);
                    for(int j=1;j<=n;j++)
                       if((base*j*j)>n)
                           break;
                       else
                           count++;
                }
        }
        System.out.print(count);
    }
    
    public static int nextNum(int num)
    {
        int count=1;
        int temp=0;
        while((count*count)<=num)
        {
            if(num%(count*count)==0)
                temp=num/count/count;
            count++;
        }
        if(temp==0)
            return num;
        else
            return temp;
    }
}

k=0
for i in range(10):
    for j in range(10):
        if (i*j)**0.5==int((i*j)**0.5):
            print(i,j)
            k=k+1
            print(k)

使用Python运行超时，用时2000ms