模拟笔试 爱奇艺2018秋招Java开发

最后一道编程题
【平方根问题】
【题目】
链接：https://www.nowcoder.com/questionTerminal/6524263a0cc9427ebb0af10b3f366774
来源：牛客网
考虑定义在两正整数上的函数SSR(平方根之和的平方):SSR(A, B) = (sqrt(A) + sqrt(B))^2。牛牛对函数值为整数的情况很感兴趣。现在给定整数n和m,请帮助牛牛计算有序对(A, B)的数量, 满足1 ≤ A ≤ n, 1 ≤ B ≤ m而且SSR(A, B)是一个整数。
输入描述:
输入包括两个整数n和m(1 ≤ n ≤ 10^5, 1 ≤ m ≤ 10^5)
输出描述:
输出一个整数,表示满足条件的有序对对数。
示例1
输入
3 8
输出
5
【解答】最开始的解是这样的

import java.util.Scanner;

/**
 * Created by Administrator on 2018/3/24 0024.
 */
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int m = scanner.nextInt();
        int big = m > n ? m : n;
        int small = m < n ? m : n;
        int sum = 0;
        for (int i = 1; i <= small; i++) {
            int tmp = 0;
            for (int j = 1; j <= big; j++) {
                if (i * j * j <= small) {
                    if (i == i * j * j) {
                        tmp++;
                    } else {
                        tmp+= 2;
                    }
                } else if (i *j * j <= big) {
                    tmp++;
                } else {
                    break;
                }
            }
            sum += tmp;
        }
        System.out.println(sum);
    }

对于m = 3, n = 8举例
【点评】从小的值开始遍历，发现这样的规律如果small值取：
1 那么大值可以取1*1*1, 1*2*2, 1*3*3,…直到<= big值
2 那么大值可以取2*1*1, 2*2*2, 2*3*3 … 直到 <= big值
3同理
但是这样只能AC 30%，漏掉情况了
如果m = 12, n =12
比如说如果small取12，那么用我的方法只有一个12 * 12的解，其实还是有别的解，就是 12 * 3
怎么解决这个问题，就用改进上面的方法，分析如果small取12， 12可以拆分成 2 * 2 *3，那么只要big中有一个因子3，比如可以取3，3*2*2，如果big值允许，还可以取3*3*3，3*4*4直到<= big。所以如果取到m=12时，我们要先求12有没有因子是完全平方的，如果有，找出因子中最大的完全平方，然后相除计算得到k，这个k就是说明取到的small值是最大平方和因子的k倍。那么在big中还可以取k*1*1, k*2*2, k*3*3….保持<= big值。

import java.util.Scanner;

/**
 * Created by Administrator on 2018/3/25 0025.
 */
public class Main1 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int m = scanner.nextInt();
        int n = scanner.nextInt();
        int big = m > n ? m : n;
        int small = m < n ? m : n;
        int res = 0;
        for (int i = 1; i <= small; i++) {
            int s = 1;
            for (int j = 2; j <= i / j; j++) {
                if (i % (j * j) == 0) {
                    s = j * j;
                }
            }
            int r = i / s;
            for (int j = 1; j * j * r <= big; j++) {
                res++;
            }
        }
        System.out.println(res);
    }
}

两种解法对于12 12的例子
1 1
1 4
4 1 //此处对于4是完全平方，解法一的big只能取1，其实还可以去4， 9
1 9
9 1 //此处对于9是完全平方，解法一种的big只能取1，其实还可以取4，即使通过交换顺序res获得了两个增量，但是前面的4已经有漏掉解了
2 2
3 3
4 4
5 5
6 6