描述
一个数列包含n(1<n<=100000)个元素,这些元素可能相同 对这个数列有m(1<=m<=100000)个询问, 每个询问包含一个序号x,请回答数列中数值小于第x个元素的元素的总和
Input
第一行包含一个数t代表总的测试样例数 每组样例包含4行 第一行包含一个整数n代表数列的长度 第二行包含n个小于1000的正整数用空格隔开 第三行包含一个整数m代表询问的个数 第四行包含m个在1到n之间的整数代表询问序号
Output
每组输出结果包含m个由空格隔开的整数代买m次询问的结果
Example Input
2 4 255 139 58 412 5 1 1 2 3 3 5 427 563 289 203 326 4 3 1 4 4
Example Output
197 197 58 0 0 203 818 0 0
解析:这道题不难,主要是有容易考虑不到的地方,我是通过结构体来存放数值和编号,然后以数值的大小排序,接着就计算每一个编号的结果值存放到res数组中,容易出错的是没有考虑数值可能重复的情况
比如这组数据
5
20 21 20 21 20
5
1 2 3 4
输出
0 60 0 60 0
如果考虑了这组数据应该问题就不大了。
贴一下自己AC的代码
#include<stdio.h>
#include<algorithm>
using std::sort;
//定义全局变量
const int M=100000+5;
struct data{
int n;
int num;
}a[M];
int b[M];
int res[M];
//优先按照数值的大小进行排序,如果数值相等再按照编号升序
bool cmp(data x,data y)
{
if(x.n<y.n)
{
return true;
}else if(x.n>y.n)
{
return false;
}else{
return x.num<y.num;
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n,cnt=1;
scanf("%d",&n);
for(int i=0; i<n; ++i)
{
scanf("%d",&a[i].n);
a[i].num=i+1;
}
int m;
scanf("%d",&m);
//输入询问的序号
for(int i=0; i<m; ++i)
{
scanf("%d",&b[i]);
}
//对数组a中的数据进行排序
sort(a,a+n,cmp);
//统计小于每个序号所对应值之和
res[a[0].num]=0;
for(int i=1; i<n; ++i)
{
if(a[i].n!=a[i-1].n)
{
res[a[i].num]=res[a[i-1].num]+a[i-1].n;
int temp=i-1;
//如果前面全是相同元素
while(temp>0&&a[i-1].n==a[temp-1].n)
{
res[a[i].num]+=a[temp-1].n;
--temp;
}
}else{
//与前面的元素相同
res[a[i].num]=res[a[i-1].num];
}
}
//输出结果
for(int i=0; i<m; ++i)
{
printf("%d",res[b[i]]);
//输出空格,最后一个数字则输出换行
if(i!=m-1)
{
printf(" ");
}else{
printf("\n");
}
}
}
}