序列查询问题描述
【题目背景】
西西艾弗岛的购物中心里店铺林立,商品琳琅满目。为了帮助游客根据自己的预算快速选择心仪的商品,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,其中最大的数为 Ai,其下标为 i,故 f(x) = i。
An ≤ x
此时序列 A 中所有的数都小于等于 x,其中最大的数为 An,故 f(x) = n。
令 sum(A) 表示 f(0) 到 f(N−1) 的总和,即:
对于给定的序列 A,试计算 sum(A)。
【输入形式】
从标准输入读入数据。
输入的第一行包含空格分隔的两个正整数 n 和 N。
输入的第二行包含 n 个用空格分隔的整数 A1,A2,⋯,An。
注意 A0 固定为 0,因此输入数据中不包括 A0。
【输出形式】
输出到标准输出。
仅输出一个整数,表示 sum(A) 的值。
【样例输入】
3 10
2 5 8
【样例输出】
15
【样例说明】
如上表所示,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): sum(A)=f(0)×2+f(2)×3+f(5)×3+f(8)×2
【样例输入】
9 10 1 2 3 4 5 6 7 8 9
【样例输出】
45
【样例说明】
若存在区间 [i,j) 满足 f(i)=f(i+1)=⋯=f(j−1),使用乘法运算 f(i)×(j−i) 代替将 f(i) 到
f(j−1) 逐个相加,或可大幅提高算法效率。
代码实现
int main()
{
int a, b = 0;
scanf("%d %d", &a, &b);
int* parr = (int*)calloc(b, 4);
int* parr2 = (int*)calloc(a, 4);
int* ret = parr;
int* rret = parr;
if (parr == NULL)
{
return 1;
}
if (parr2 == NULL)
{
return 1;
}
int* ret2 = parr2;
int count = 0;
while (1)//输入数据
{
scanf("%d", ret2+ count);
count++;
if (count == a)
{
break;
}
}
int uint = 1;
int j = 0;
int m = 0;
int ci = 0;
while (1)//第一次循环 将parr初始化 构造 f(i)
{
for (m = 0; m < a; m++)
{
*(ret + *(ret2)) = uint;
int* mid = (ret + *(ret2));
for (j = 0; j < (b - *(ret2)-1); j++)
{
mid++;
*mid = uint;
}
ci++;
if (ci == a)
{
break;
}
ret2++;
uint++;
ret = rret;
}
break;
}
int sum = 0;
for (m = 0; m < b; m++)
{
sum += *(rret + m);
}
printf("%d", sum);
free(parr);
free(parr2);
ret = NULL;
ret2 = NULL;
rret = NULL;
return 0;
}
总结与解释
- 开辟两块数组,第一块存放我们所输入的A1 A2 A3…,第二块则用来存放f(i)。
接下来用一次双循环首先,根据第一块数组内的数字将它作为下标,用来访问第二块数组,并将改元素及其后续所有元素初始化为1.接着根据第一块数组的第二个数据作为下标,用来访问第二块数组,并将改元素及其后续所有元素初始化为2.以此循环,我们便可以构造出f(i)。最后求和即可。