bzoj 2093 [Poi2010]Frog——滑动窗口

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=2093

找第k近的可以用一个含k个元素的滑动窗口来实现。

卡空间也还行，但卡时间。不要预处理倍增数组再计算，开一维，一边处理倍增数组一边算答案。不要每次算M次，for(;m;m>>=1)更快。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e6+,M=;
int n,K,hd,tl,to[N],tt[N],ans[N];
ll m,dis[N];
int rdn()
{
    int ret=;char ch=getchar();
    while(ch>''||ch<'')ch=getchar();
    while(ch>=''&&ch<='')(ret*=)+=ch-'',ch=getchar();
    return ret;
}
ll rdl()
{
    ll ret=;char ch=getchar();
    while(ch>''||ch<'')ch=getchar();
    while(ch>=''&&ch<='')(ret*=)+=ch-'',ch=getchar();
    return ret;
}
int main()
{
    n=rdn();K=rdn();m=rdl();K++; //zj是第一近
    for(int i=;i<=n;i++)dis[i]=rdl();
    hd=;tl=K;
    for(int i=;i<=n;i++)
    {
        ans[i]=i;
        while(tl<n&&dis[tl+]-dis[i]<dis[i]-dis[hd])hd++,tl++;
        to[i]=(dis[tl]-dis[i]>dis[i]-dis[hd]?tl:hd);
    }
    if(m&)
        for(int i=;i<=n;i++)ans[i]=to[ans[i]];
    m>>=;
    for(;m;m>>=)
    {
        for(int j=;j<=n;j++) tt[j]=to[to[j]];
        if(m&) for(int j=;j<=n;j++) ans[j]=tt[ans[j]];
        for(int j=;j<=n;j++) to[j]=tt[j];
    }
    for(int i=;i<=n;i++)printf("%d ",ans[i]);
    return ;
}