写在前面:
介个题呢是某THU大神,也是我原来的少儿部同学MQC MM,鉴于她所在自动化系的大作业需要这个题,我应邀写了这篇质量不高的解题报告。
POI XVII frog(zab)解题报告
首先,我必须承认我这题SB了……经历了将近6个小时的编写和调试,发现了一个异常悲剧的情况:我第一个算法果断超时了……。虽然是超时,但是超的非常少,几乎是擦边,优化了一下午的常数结果是我放弃了……转而去想我刚开始就否定的算法,结果15分钟编写加调试,居然 TMD过了……言归正传,说说这个题的解法:
首先说说这个题的题意:说在一条小溪上有n个石头,按照严格升序给出每个是石头到源头的距离。然后说有一群灰常SB的青蛙(每块石头上一只),每次它们会选择距离自己第k远的石头跳过去。(第i个石头和第j个石头的距离为abs(a[i] – a[j])),如果距离当前位置第k远的石头不止一个,那就调到距离源头更近的那个石头。问跳m次之后,最开始处在i位置上的青蛙最终会跳到哪个石头上。
数据范围:1<k<n<=106, m<=1018, a[i] < 1018
看到数据范围基本已经确定这道题的复杂度了,基本上就是O(n)或者是O(nlogn)的了。接下来我们来拆解这个问题。如果我们知道了处在i位置的青蛙第一次会跳到哪个石头上,我们就可以分治的解决这个问题,从而求出第m次青蛙在哪了。具体的想法是这样的:因为如果青蛙的位置确定了,那么它下一次调到的位置也就确定了。所以我们可以通过求出第2^k次时青蛙跳到的位置,然后将m拆解成二进制来求。这样这部分的算法复杂度确定了就是O(nlogm)。其实也能构个有向图,然后强连缩点外带标号(其实是个外向树),能做到线性的复杂度,但是真的没啥用(能过得了)。
接下来就是最悲剧的地方:如何求出距离一个点第k远点的位置。首先直接循环统计肯定是要超时的,接着我就想到了二分。对于任意一个点x,实际上答案在1~x-1和x+1~n这两个区间上都是又单调性的,我最开始的想法就是在两侧分别进行二分,然后check的时候加上另一侧比这个数小的数的个数,这个当然也要二分。也就是两次二分,复杂度就是O(n*logn*logn)的。但是很不幸,这个算法是超时的,问题就出在这两个悲剧的log上。虽然不是很大,但是也比线性慢了100倍左右。就此调了N久,经过各种大神的指点,发现其实被我最开始否定的线性算法是对的……
这个算法是这样的:维护一个类似队列的东西,但是实际上就是维护两个指针front和rear, 并且保证front – rear 始终等于k,即front ~ rear始终有k + 1个数。这里编号从0开始。Data[0](距离第0个点第k远的点的位置)肯定是k,这个没有问题。接着很容易得到初始化边界front = 0, rear = k;接着开始处理data[1],data[2]……以此类推。Data[i]处理结束后,接着处理data[I + 1],这时front和rear就要有改变了,我们比较最左边的点(front)和可能要加入的下一个点(rear + 1)距离当前点的距离,如果rear + 1相对当前点的距离比front要近,那么将front和rear组成的这个边界向右移动一格(front++, rear++)。一旦front和rear确定了(不能再动了),我们发现其实这k+1个数就是距离当前点最近的k+1数(包括自身)。这样我们就可以直接比较front和rear相对当前点的距离了,距离较远的就是答案,如果相等,就取左边的。至此,这个问题就解决了,因为每个点都至多进队一次出队一次,所以复杂度是线性的。
综上:我原来的带两个log的算法超时并且爆长……写的无限丑陋。之后的标程只有不到70行,产生了巨大的反差……总之一句话:动手之前一定要想明白啊……
P.s. MQC童鞋,那个说实话,这里真没用啥数据结构……您可以再问问怎么用的数据结构……
以下为该题目的源码:
#include <iostream>
#include <fstream>
#include <cstring>
using namespace std;
long long a[1000001];
int ans[1000001];
int data[1000001], tmp[1000001];
long long m;
int n, k;
void init()
{
scanf("%d%d%lld", &n, &k, &m);
for(int i = 0; i < n; i++)
scanf("%lld", &a[i]);
}
void solve()
{
int front = 0, rear = k;
data[0] = k;
for (int i = 1; i < n; i++)
{
while (rear < n - 1 && a[i] - a[front] > a[rear + 1] - a[i])
{
front++; rear++;
}
if (a[i] - a[front] >= a[rear]- a[i]) data[i] = front;
else data[i] = rear;
}
for (int i = 0; i < n; i++)
ans[i] = i;
while (m > 0)
{
if (m % 2 == 1)
for (int i = 0; i < n; i++)
ans[i] = data[ans[i]];
memcpy(tmp, data, sizeof(data));
for (int i = 0; i < n; i++)
data[i] = tmp[tmp[i]];
m /= 2;
}
}
void print()
{
printf("%d", ans[0] + 1);
for (int i = 1; i < n; i++)
printf(" %d", ans[i] + 1);
printf("\n");
}
int main()
{
freopen("zab.in", "r", stdin);
freopen("zab.out", "w", stdout);
init();
solve();
print();
return 0;
}