题目背景
西西艾弗岛的购物中心里店铺林立,商品琳琅满目。为了帮助游客根据自己的预算快速选择心仪的商品,IT 部门决定研发一套商品检索系统,支持对任意给定的预算 𝑥,查询在该预算范围内(≤𝑥)价格最高的商品。如果没有商品符合该预算要求,便向游客推荐可以免费领取的西西艾弗岛定制纪念品。
假设购物中心里有 𝑛 件商品,价格从低到高依次为 𝐴1,𝐴2⋯𝐴𝑛,则根据预算 𝑥 检索商品的过程可以抽象为如下序列查询问题。
题目描述
𝐴=[𝐴0,𝐴1,𝐴2,⋯,𝐴𝑛] 是一个由 𝑛+1 个 [0,𝑁) 范围内整数组成的序列,满足 0=𝐴0<𝐴1<𝐴2<⋯<𝐴𝑛<𝑁。(这个定义中蕴含了 𝑛 一定小于 𝑁。)
基于序列 𝐴,对于 [0,𝑁) 范围内任意的整数 𝑥,查询 𝑓(𝑥) 定义为:序列 𝐴 中小于等于 𝑥 的整数里最大的数的下标。具体来说有以下两种情况:
存在下标 0≤𝑖<𝑛 满足 𝐴𝑖≤𝑥<𝐴𝑖+1
此时序列 𝐴 中从 𝐴0 到 𝐴𝑖 均小于等于 𝑥,其中最大的数为 𝐴𝑖,其下标为 𝑖,故 𝑓(𝑥)=𝑖。
𝐴𝑛≤𝑥
此时序列 𝐴 中所有的数都小于等于 𝑥,其中最大的数为 𝐴𝑛,故 𝑓(𝑥)=𝑛。
令 𝑠𝑢𝑚(𝐴) 表示 𝑓(0) 到 𝑓(𝑁−1) 的总和,即:
𝑠𝑢𝑚(𝐴)=∑𝑖=0𝑁−1𝑓(𝑖)=𝑓(0)+𝑓(1)+𝑓(2)+⋯+𝑓(𝑁−1)
对于给定的序列 𝐴,试计算 𝑠𝑢𝑚(𝐴)。
输入格式
从标准输入读入数据。
输入的第一行包含空格分隔的两个正整数 𝑛 和 𝑁。
输入的第二行包含 𝑛 个用空格分隔的整数 𝐴1,𝐴2,⋯,𝐴𝑛。
注意 𝐴0 固定为 0,因此输入数据中不包括 𝐴0。
输出格式
输出到标准输出。
仅输出一个整数,表示 𝑠𝑢𝑚(𝐴) 的值。
样例1输入
3 10
2 5 8
Data
样例1输出
15
Data
样例1解释
𝐴=[0,2,5,8]
𝑖 0 1 2 3 4 5 6 7 8 9
𝑓(𝑖) 0 0 1 1 1 2 2 2 3 3
如上表所示,𝑠𝑢𝑚(𝐴)=𝑓(0)+𝑓(1)+⋯+𝑓(9)=15。
考虑到 𝑓(0)=𝑓(1)、𝑓(2)=𝑓(3)=𝑓(4)、𝑓(5)=𝑓(6)=𝑓(7) 以及 𝑓(8)=𝑓(9),亦可通过如下算式计算 𝑠𝑢𝑚(𝐴):
𝑠𝑢𝑚(𝐴)=𝑓(0)×2+𝑓(2)×3+𝑓(5)×3+𝑓(8)×2
样例2输入
9 10
1 2 3 4 5 6 7 8 9
Data
样例2输出
45
Data
子任务
50% 的测试数据满足 1≤𝑛≤200 且 𝑛<𝑁≤1000;
全部的测试数据满足 1≤𝑛≤200 且 𝑛<𝑁≤107。
提示
若存在区间 [𝑖,𝑗) 满足 𝑓(𝑖)=𝑓(𝑖+1)=⋯=𝑓(𝑗−1),使用乘法运算 𝑓(𝑖)×(𝑗−𝑖) 代替将 𝑓(𝑖) 到 𝑓(𝑗−1) 逐个相加,或可大幅提高算法效率。
代码:
#include <iostream>
using namespace std;
int main()
{
int n,m;
cin>>n>>m;
int a[200]={0};
int sum=0;
for(int i=1;i<=n;i++){
cin>>a[i];
}
for(int i=1;i<n;i++){
sum=sum+i*(a[i+1]-a[i]);
}
sum=sum+n*(m-a[n]);
cout<<sum<<endl;
}