202112-1

202112-1
ccf csp认证
试题编号： 202112-1
试题名称： 序列查询
时间限制： 300ms
内存限制： 512.0MB
问题描述：
题目背景
西西艾弗岛的购物中心里店铺林立，商品琳琅满目。为了帮助游客根据自己的预算快速选择心仪的商品，IT 部门决定研发一套商品检索系统，支持对任意给定的预算 x，查询在该预算范围内（ ≤ x）价格最高的商品。如果没有商品符合该预算要求，便向游客推荐可以免费领取的西西艾弗岛定制纪念品。

假设购物中心里有 n 件商品，价格从低到高依次为 A1, A2, …, An，则根据预算 x 检索商品的过程可以抽象为如下序列查询问题。

题目描述
A=|A0,A1, A2, …, An| 是一个由 n + 1 个 [0, N) 范围内整数组成的序列，满足 0 = A0 < A1 < A2 < … < An < N。（这个定义中蕴含了 n 一定小于 N。）

基于序列 A，对于 [0, N) 范围内任意的整数 x，查询 f(x) 定义为：序列 A 中小于等于 x 的整数里最大的数的下标。具体来说有以下两种情况：

存在下标 0 ≤ i < n 满足 Ai ≤ x < Ai+1
此时序列 A 中从 A0 到 Ai 均小于等于 x，其中最大的数为 An，其下标为 i，故 f(x) = i。
An ≤ x
此时序列 A 中所有的数都小于等于 x，其中最大的数为 An，故 f(x) = n。
令 sum(A) 表示 f(0) 到 f(N - 1) 的总和，即：
s u m ( A ) = ∑ i = 0 N − 1 f ( i ) = f ( 0 ) + f ( 1 ) + f ( 2 ) + . . . + f ( N − 1 ) sum(A)=\sum_{i=0}^{N-1}{f(i) = f(0) +f(1) + f(2) + … + f(N-1)}
sum(A)=
i=0
∑
N−1
​
f(i)=f(0)+f(1)+f(2)+…+f(N−1)

对于给定的序列 A，试计算 sum(A)。
输入格式
从标准输入读入数据。
输入的第一行包含空格分隔的两个正整数 n 和 N。
输入的第二行包含 n 个用空格分隔的整数 A1, A2, … , An。
注意 A0 固定为 0，因此输入数据中不包括 A0。

输出格式
输出到标准输出。
仅输出一个整数，表示 sum(A) 的值。

样例1输入
3 10
2 5 8
样例1输出
15
样例1解释
A = {0, 2, 5, 8}
0 1 2 3 4 5 6 7 8 9
f(x) 0 0 1 1 1 2 2 2 3 3
如上表所示，sum(A) = f(0) + f(1) + … + f(9) = 15。
考虑到 f(0) = f(1)、f(2) = f(3) = f(4) 、f(5) = f(6) = f(7) 以及 f(8) = f(9)，亦可通过如下算式计算 sum(A) ：
s u m ( A ) = f ( 0 ) × 2 + f ( 2 ) × 3 + f ( 5 ) × 3 + f ( 8 ) × 2 sum(A) = f(0) × 2 + f(2) × 3 + f(5) × 3 + f(8) × 2
sum(A)=f(0)×2+f(2)×3+f(5)×3+f(8)×2

样例2输入
9 10
1 2 3 4 5 6 7 8 9
样例2输出
45

子任务
50% 的测试数据满足 1 ≤ n ≤ 200 且 n < N ≤ 1000；
全部的测试数据满足 1 ≤ n ≤ 2000 且 n < N ≤ 107。

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

int main() {
	int n;
	int N;
	cin>>n;
	cin >> N;
	vector<int> a(n+1);
	a[0] = 0;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	vector<int> fi(N+1);
	int m = 1;
	int flag = 0;
	for (int i = 0; i < N; i++) {
		if (a[m] == i) {
			flag++;
			if (m < n) {
				m++;
			}
		}
		fi[i] = flag;
	}
	int sum = 0;
	for (int i = 0; i < N; i++) {
		sum = sum + fi[i];
	}
	cout << sum;

}