【题解】LuoGu3509:[POI2010]ZAB-Frog

原题传送门
m < = 1 0 18 m<=10^{18} m<=1018,想到倍增
先预处理出每个点跳一次到达的点
倍增过程参考快速幂方法

现在考虑如何求出每个点跳一次会跳到那个点
一般会想到直接暴力套几个数据结构,比如主席树之类的
但是我们朴素一点,可以用单调队列思想

  • 记两个头尾指针 l , r ( r − l = k ) l,r(r-l=k) l,r(rl=k),框定当前点距离最小的 k k k个点
  • 初始时 l = 1 , r = k + 1 l=1,r=k+1 l=1,r=k+1
  • 每次枚举的点往后移一位时,把 r + 1 与 l r+1与l r+1l到当前点的距离进行比较,类似滑动窗口
  • 最终 l , r l,r l,r中距离当前点更大的那个就是答案

Code:

#include 
#define maxn 1000010
#define LL long long
using namespace std;
int n, k, st[maxn][2], ans[maxn];
LL m, p[maxn];

inline LL read(){
	LL s = 0, w = 1;
	char c = getchar();
	for (; !isdigit(c); c = getchar()) if (c == '-') w = -1;
	for (; isdigit(c); c = getchar()) s = (s << 1) + (s << 3) + (c ^ 48);
	return s * w;
}

int main(){
	n = read(), k = read(), m = read();
	for (int i = 1; i <= n; ++i) p[i] = read();
	int l = 1, r = k + 1;
	for (int i = 1; i <= n; ++i){
		ans[i] = i;
		while (r < n && p[r + 1] - p[i] < p[i] - p[l]) ++l, ++r;
		if (p[r] - p[i] > p[i] - p[l]) st[i][0] = r; else st[i][0] = l;
	}
	int tmp;
	for (int j = 1; m; m >>= 1, j ^= 1)
		for (int i = 1; i <= n; ++i){
			if (m & 1) ans[i] = st[ans[i]][j ^ 1];
			st[i][j] = st[st[i][j ^ 1]][j ^ 1];
		}
	for (int i = 1; i <= n; ++i) printf("%d ", ans[i]);
	return 0;
}

你可能感兴趣的:(题解,LuoGu,倍增)