标签:ios 逆时针 tail accept make == out ons print
环上有 n 个点,按顺时针顺序以 1 到 n 编号。其中 1~m 号点是红色的,m+1~n 号点时蓝色的。
一开始你位于点 s,并给定一个 t。
你需要重复以下步骤,直到 t = 0:
如果你所在结点为红色,顺时针移动 t 个点;否则逆时针移动 t 个点。然后 t 减一。
求最终所在的结点编号。
Input
第一行包含两个整数 n 和 m (3≤n≤10^5, 1≤m<n),含义见上。
第二行包含两个整数 s 和 t (1≤s≤n, 1≤t≤10^12),含义见上。
Output
仅输出一个整数,表示最终所在的结点编号。
Examples
Input
10 4
3 1
Output
4
Input
10 4
3 5
Output
4
不难想到当 x > n 时,我们等价于走 x mod n 步(n 步相当于绕一个圈子)。
于是我们可以预先走 t mod n 步,之后每一次就走 n 步(即走 n, n - 1, ..., 1 步),重复 t / n 次。
我们可以预处理每个点走 n, n - 1, ..., 1 步所到达的目的地,然后就可以倍增了。
接着考虑怎么预处理每个点走 n, n - 1, ..., 1 步所到达的目的地。我们不妨大胆设一个 dp。
假如说正着设 dp 状态(即定义 dp[i][j] 表示从 i 出发走 n, ..., n - j 步所到达的目的地),我们的转移依赖于 dp[i][j-1],不好进行优化。
于是我们倒着设 dp 状态,即定义 dp[i][j] 表示从 i 出发走 j, ..., 1 步所到达的目的地。
这样子 dp[i][j] 在 i <= m 时等于 dp[i+j][j-1];在 i > m 时等于 dp[i-j][j-1](这里的 ±j 是在环的意义下进行加减)。
然后我们就可以优化这个 dp 了:注意它的转移总是 dp[][j-1] 的两个个区间进行平移得到 dp[][j]。
于是我们使用可持久化 treap,每次 dp 转移相当于先分裂出对应区间,再将区间合并。
有一个小 trick:可持久化 treap 中深度可能比较深,可以尝试将 merge 写成随机合并 + 定期重构(当然也不一定要加。。。这个看具体的人而定)。
#include<cstdio>
#include<cstdlib>
#include<ctime>
#include<iostream>
using namespace std;
typedef long long ll;
const int MAXN = 100000;
struct treap{
struct node{
int key, siz;
node *ch[2];
}pl[30*MAXN + 5], *ncnt, *NIL;
typedef pair<node*, node*> Droot;
treap() {
NIL = &pl[0];
NIL->ch[0] = NIL->ch[1] = NIL;
NIL->key = NIL->siz = 0;
}
void trans(int *f, node *x, int k) {
if( x == NIL ) return ;
f[x->ch[0]->siz + k + 1] = x->key;
trans(f, x->ch[0], k);
trans(f, x->ch[1], x->ch[0]->siz + k + 1);
}
void pushup(node *x) {
x->siz = x->ch[0]->siz + x->ch[1]->siz + 1;
}
node *build(int *f, int l, int r) {
if( l > r ) return NIL;
int mid = (l + r) >> 1;
node *p = (++ncnt); p->key = f[mid];
p->ch[0] = build(f, l, mid - 1);
p->ch[1] = build(f, mid + 1, r);
pushup(p);
return p;
}
node *build(int *f, int n) {
ncnt = &pl[0];
return build(f, 1, n);
}
Droot split(node *x, int k) {
if( x == NIL ) return make_pair(NIL, NIL);
node *p = (++ncnt); (*p) = (*x);
if( x->ch[0]->siz >= k ) {
Droot q = split(x->ch[0], k);
p->ch[0] = q.second; pushup(p);
return make_pair(q.first, p);
}
else {
Droot q = split(x->ch[1], k - x->ch[0]->siz - 1);
p->ch[1] = q.first; pushup(p);
return make_pair(p, q.second);
}
}//first k elements
node *merge(node *x, node *y) {
if( x == NIL ) return y;
if( y == NIL ) return x;
node *p = (++ncnt);
if( rand() & 1 ) {
(*p) = (*x), p->ch[1] = merge(x->ch[1], y);
pushup(p);
}
else {
(*p) = (*y), p->ch[0] = merge(x, y->ch[0]);
pushup(p);
}
return p;
}
}T;
treap::node *root;
int f[MAXN + 5];
int nxt[45][MAXN + 5];
int main() {
int n, m, s; ll t;
scanf("%d%d%d%lld", &n, &m, &s, &t);
srand(20041112 ^ n ^ m ^ s);
for(int i=1;i<=n;i++) f[i] = i;
root = T.build(f, n);
for(int i=1;i<=n;i++) {
if( i % 300 == 0 )
T.trans(f, root, 0), root = T.build(f, n);
treap::node *p, *q;
treap::node *tmp = T.split(root, i).second;
if( tmp->siz >= m )
p = T.split(tmp, m).first;
else p = T.merge(tmp, T.split(root, m-tmp->siz).first);
tmp = T.split(root, n - i).first;
if( tmp->siz >= n-m )
q = T.split(tmp, tmp->siz-(n-m)).second;
else q = T.merge(T.split(root, n-((n-m)-tmp->siz)).second, tmp);
root = T.merge(p, q);
}
T.trans(f, root, 0);
while( t % n ) {
if( s <= m ) s = (s - 1 + t) % n + 1;
else s = ((s - 1 - t) % n + n) % n + 1;
t--;
}
for(int i=1;i<=n;i++) nxt[0][i] = f[i];
for(int j=1;j<45;j++)
for(int i=1;i<=n;i++)
nxt[j][i] = nxt[j-1][nxt[j-1][i]];
int x = t/n;
for(int i=44;i>=0;i--)
if( (1LL<<i) & x ) s = nxt[i][s];
printf("%d\n", s);
}
定时重构。。。感觉非常骚。。。
当然这是基于 treap 本身内部结点个数是恒定的前提下。
因为有定时重构的缘故所以空间也不需要消耗特别大。
@codeforces - 1056G@ Take Metro
标签:ios 逆时针 tail accept make == out ons print
原文地址:https://www.cnblogs.com/Tiw-Air-OAO/p/11688924.html