序列查询新解

序列查询新解

题目背景

上一题“序列查询”中说道：A=[A₀,A₁,A₂···,A_n] 是一个由 n+1 个 [0,N] 范围内整数组成的序列，满足 0=A₀<A₁<A₂<···<A_n<N。基于序列 A ，对于 [0,N] 范围内任意的整数 x，查询 f(x) 定义为：序列 A 中小于等于 x 的整数里最大的数的下标。

对于给定的序列 A 和整数 x，查询 f(x) 是一个很经典的问题，可以使用二分搜索在 O(log n) 的时间复杂度内轻松解决。但在 IT 部门讨论如何实现这一功能时，小 P 同学提出了些新的想法。

题目描述

输入格式

从标准输入读入数据。

输入的第一行包含空格分隔的两个正整数 n 和 N。

输入的第二行包含 n 个用空格分隔的整数A₁,A₂,···,A_N。

注意 A₀ 固定为 0，因此输入数据中不包括 A₀。

输出格式

输出到标准输出。

仅输出一个整数，表示 error(A) 的值。

样例1输入

3 10
2 5 8

样例1输出

5

样例1解释

样例2输入

9 10
1 2 3 4 5 6 7 8 9

样例2输出

0

样例3输入

2 10
1 3

样例3输出

6

样例3输出

子任务

提示

需要注意，输入数据 [A₁,A₂···,A_n] 并不一定均匀分布在 (0,N) 区间，因此总误差 error(A) 可能很大。

代码

#include <bits/stdc++.h>
using namespace std;
long long n, N;
const long long Maxn = 1e5 + 5;
long long arr[Maxn] = { 0 };
long long vi[Maxn] = { 0 };
long long r;
long long num = 0;
long long add(long long first, long long count) {//first表示第一位的g(x)-f(x),count代表该区间包含了几个均分区间
    long long last = first + count - 1;//末项，该区间处于一个均分区间中的话last就等于2*first
    long long final;
    if (last * first >= 0)//同符号，表示起点和终点在同一个均分区间内
        return final = abs((first + last) * count / 2 * r);//也就是计算了起点到结束的所有的绝对值之和，*r就代表了区间内的所有和
    else//不同符号，first是负数 表示两个点不在同一均分区间内
        return final = (abs(first) + 1) * abs(first) / 2 * r + last * (last + 1) / 2 * r;//分开计算正数和负数
}
int main()
{
    cin >> n >> N;
    arr[0] = 0;
    long long lea = 0;//表示最终结果
    long long co,k;
    r = N / (n + 1);
    for (long long i = 1; i <= n; i++) {
        cin >> arr[i]; 
        vi[i] = arr[i] - arr[i - 1];//用vi数组存储区间的长度
    }
    vi[n+1]= N - arr[n];//最后存储大于最后读入的数据的部分的区间长度
    for (long long i = 1; i <= n+1; i++) {
        co = (vi[i]+ lea)/r;//计算当前区间相当于多少个均分区间
        k = arr[i - 1] / r;//计算区间开始的第一个元素的g(x)对应的值
        num += add(k - i + 1, co);//计算并加上该区间的error值
        if(k-i+1>=0)num += lea;//如果当前的区间的g(x)>f(x)，则加上均分区间多出来的数字个数
        else num -= lea;//否则当前区间减去减去均分区间多出来的数字个数
        lea = (vi[i] + lea) % r;//lea表示不在前部分均分区间的数字个数
    }
    k += co;
    num += abs(k - n) * lea;//最后加上g(x)与f(x)的差值的绝对值乘以剩余的没计算的数字个数
    cout<<num;
    return 0;
}

思路

思路就是先将题目中的r计算出来，模拟将[0,N]的区段分为 N/r 个均分的区间，在对应的区间里所有的数字对应的g(x)的值都相同。然后就是通过读入的数组来对[0,N]的区段进行另一种分段，这次的分段决定f(x)，将相同的f(x)的数字分为一个区间。然后就是每次取第二种分段的一段，判断区间的左右端以及是否属于同一均分区间，进而判断区间中g(x)与f(x)的差值的正负，然后计算该区间的差值的绝对值的和，最后循环计算即可。