调和序列“盛大游戏杯”第15届上海大学程序设计联赛夏季赛暨上海高校金马五校赛

调和序列

发布时间: 2017年7月9日 20:20   最后更新: 2017年7月10日 21:11   时间限制: 1000ms   内存限制: 128M

描述

给定一个长度为n的非负整数序列，下标为0,1,…,n−1. 

定义：sequence(K)： 由下标为K的倍数组成的子序列,即下标为0,K,2K,...,[n−1/k]∗k

query(K,S): 询问sequence(K)中的第S大的数字

输入

第一行一个整数T，表示测试组数。
对于每组数据，第一行输入两个整数n,m，1<=n<=20000, 1<=m<=100000，n表示序列的长度，m表示询问个数。
接下来一行是n个整数a0,a1,..,an−1,0<=ai<231, i=0,1,…,n−1，表示序列。
接下来m行，每行两个整数K,S 
0<K<=109, 1<=S<=n

输出

每组数据对于每个询问输出一行，若sequence(K)的元素个数小于S，输出−1；否则输出query(K,S)

样例输入1 复制

15 22 5 3 4 12 42 1

样例输出1

-1 3

选择语言

想法：

使用<vector>并进行优化(使用sort函数排序)

代码：
#include<bits/stdc++.h>
using namespace std;
const int maxn= 20005;
vector<int>vec[maxn];
int a[maxn];
bool flag[maxn];
int n,m;
void init()
{
    for(int i=1;i<n;i++)
    {
        for(int j=0;j<n;j+=i)
            vec[i].push_back(a[j]);
    }
}
int main()
{
   int T;
   scanf("%d",&T);
   while(T--)
    {
        memset(flag,0,sizeof(flag));
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
            vec[i].clear();
        }
        init();
        int k,s;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&k,&s);
            if(k>=n)
            {
                if(s==1) printf("%d\n",a[0]);
                else puts("-1");
                continue;
            }
            if(vec[k].size()<s)    //优化1
            {
                puts("-1");
                continue;
            }
            if(flag[k]==0)         //优化2
            {
                sort(vec[k].begin(),vec[k].end());
                flag[k]=1;
            }
            printf("%d\n",vec[k][vec[k].size()-s]);
        }
    }
    return 0;
}