java必刷题——众数选三

Posted 2021-07-17 我想月薪过万

题目

需求分析

解题思路

1. 获取用户输入的 N、D
2. 存储用户输入的可用坐标点
3. 从可取点中取出两个坐标

技术栈

排列组合

代码实现

package OpenClosedPrinciple;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Scanner;

/**
 * ClassName: OpenCloseTest <br/>
 * Description: <br/>
 * date: 2021/6/19 20:56<br/>
 *
 * @author yiqi<br />
 * @since JDK 1.8
 */
public class OpenCloseTest {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        //1、获取 可选点数N 最大距离D
        int N = scanner.nextInt();
        int D = scanner.nextInt();

        //2、存储可选点
        List<Integer> able_position = new ArrayList<>();
        while (scanner.hasNextInt()){
            able_position.add(scanner.nextInt());
            if (able_position.size() == N){
                break;
            }
        }

        //3、选点思路：确定 最左边 和 最右边，那中间的点就都可以。所以利用排列组合取点即可
        long total_way = 0;
        //因为要取三个点，所以 0、1要留着，最右边一个点从角标为2的选起
        int right = 2;
        //左边的点循环固定选取 N-2原理同上
        for (int left=0; left<N-2; left++){
            while (right<N && able_position.get(right)-able_position.get(left)<=D){
                right++;
            }
            // right-1:因为上面要跳出循环 right 多加了个1，减1得本身；
            // right-1-left：求 最右边点 到 最右边点 之间有几个可选择点
            total_way = (total_way+cWay(right-1-left))%99997867;
        }
        System.out.println(total_way);
    }

    private static long cWay(long pointNum) {
        //这一步运用了排列 C(n,m) 的公式 ,m是固定的为2，所以变动的是n 即 pointNum
        return pointNum*(pointNum-1)/2%99997867;
    }
}

注意

total_way 是 long 类型

cWay 返回的也是 long 类型