上海五校联赛 H 调和序列

中文题意不解释

一开始以为是个线段树之类的数据结构 毕竟还有查询

但是算了一下时间复杂度发现直接暴力就可以

O(n*sqrt(n)*sqrt(n)*log(sqrt(n))) 大概只有2e6的复杂度

对于每一个n对每个k≤sqrt(n)预处理排序 大于的部分直接计算就可以

反正还是很水的了

记得M是构造两棵线段树 但是没来得及做 晚上有时间的话补上

#include<bits/stdc++.h>
#define cl(a,b) memset(a,b,sizeof(a))
#define debug(a) cerr<<#a<<"=="<<a<<endl
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;

const int maxn=2e4+10;

int n,m;
int a[maxn];
vector<int>ans[205]; //也不知道会不会爆内存 就直接用vector了

bool cmp(int a,int b)
{
    return a>b;
}

void init()
{
    for(int i=1;i<=sqrt(n);i++) //预处理sqrt(n)的部分
    {
        ans[i].clear();
        for(int j=0;j<n;j+=i)
        {
            ans[i].push_back(a[j]);
        }
        sort(ans[i].begin(),ans[i].end(),cmp);
    }
}

int solve(int k,int s)
{
    if(k <= sqrt(n)) //对于≤sqrt(n)的k值直接查询
    {
        if(s>ans[k].size()) return -1;
        else return ans[k][s-1];
    }
    else //大于的直接暴力 反正也没几个数
    {
        vector<int>tmp;
        tmp.clear();
        for(int i=0;i<n;i+=k)
        {
            tmp.push_back(a[i]);
        }
        sort(tmp.begin(),tmp.end(),cmp);
        if(s>tmp.size()) return -1;
        else return tmp[s-1];
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        cl(a,0);
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
        }
        init(); //预处理
        while(m--)
        {
            int k,s;
            scanf("%d%d",&k,&s);
            printf("%d
",solve(k,s));
        }
    }
    return 0;
}/*

1
5 2
2 5 3 4 1
2 4
2 1

*/