【CSP：202112-2】序列查询新解（Java）

题目链接

• 202112-2 序列查询新解

题目描述

上题题解：【CSP：202112-1】序列查询（Java）

求解思路

• 这道题目给定的N的数据量很大，第一次提交代码的时候只写了一个关于N的for循环（$O(N)$)，就超时了，因此想要拿满分就只能写关于n的for循环。
• 根据题目我们可以得知，$f(x)$ 和 $g(x)$ 都是每次自增1的单调递增函数，因此我们可以像第一题一样，尝试去遍历 $f(x)$ 函数的每个分割点（即a[i])。
• 在每个 $f(x)$ 值相同的区间内去分情况讨论，如果 $f(x)$ 的这段区间内只有一个 $g(x)$ 值，那么直接计算即可，如果存在多个 $g(x)$ 值，则需要计算出每段 $g(x)$ 的值。
• a[i]表示 $f(x)=i-1$ 和 $f(x)=i$ 的边界分割点。
• left和right分别表示分割区间的左右边界。
• f表示当前分割点左侧 $f(x)$ 的值。
• j表示分割 $g(x)$ 区间的分割点。

实现代码

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int N = in.nextInt();
        int[] a = new int[n + 2];
        int r = N / (n + 1);
        long error = 0;

        for (int i = 1; i <= n; i++) {
            a[i] = in.nextInt();
        }
        a[++ n] = N; //补充上最后一个分割点，即f(x) = n的一段

        int left, right;

        for (int i = 1; i <= n; i ++) {
            int f = i - 1;
            left = a[i - 1];
            right = a[i] - 1;
            if (left / r != right / r) { //区间内有多个g(x)
                for (int j = left + (r - left % r - 1); j <= right; j += r) { //将区间按照g(x)不同分割开
                    error += Math.abs((left / r - f) * (j - left + 1)); //j表示分割点
                    left = j + 1; //更新left到g(x)的下一个区间边界
                }
                error += Math.abs((right / r - f) * (right - left + 1)); //处理最后一段
            } else { //如果区间内只有一段g(x)
                error += Math.abs((left / r - f) * (right - left + 1));
            }
        }
        System.out.println(error);
    }
}