又是一道比较神的题Orz。看了题解才会做。
首先对于每个点距离k大的点,我们可以O(n)的时间扫一遍求出来,这个是可做的。
看到范围10^18也会想到倍增吧……空间有点紧,开一个数组转移一下就好了。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n,k;
ll m,a[1000005];
int f[1000005],g[1000005],t[1000005];
inline ll read()
{
ll a=0,f=1; char c=getchar();
while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
return a*f;
}
int main()
{
n=read(); k=read(); m=read();
for (int i=1;i<=n;i++) a[i]=read();
int l=1,r=k+1; f[1]=r;
for (int i=2;i<=n;i++)
{
while (r<n&&a[i]-a[l]>a[r+1]-a[i]) l++,r++;
f[i]=a[i]-a[l]>=a[r]-a[i]?l:r;
}
for (int i=1;i<=n;i++) t[i]=i;
while (m)
{
if (m&1)
{
for (int i=1;i<=n;i++) g[i]=f[t[i]];
for (int i=1;i<=n;i++) t[i]=g[i];
}
m>>=1;
for (int i=1;i<=n;i++) g[i]=f[f[i]];
for (int i=1;i<=n;i++) f[i]=g[i];
}
for (int i=1;i<n;i++) printf("%d ",t[i]);
cout << t[n] << endl;
return 0;
}