某天,Lostmonkey发明了一种超级弹力装置,为了在他的绵羊朋友面前显摆,他邀请小绵羊一起玩个游戏。游戏一开始,Lostmonkey在地上沿着一条直线摆上n个装置,每个装置设定初始弹力系数ki,当绵羊达到第i个装置时,它会往后弹ki步,达到第i+ki个装置,若不存在第i+ki个装置,则绵羊被弹飞。绵羊想知道当它从第i个装置起步时,被弹几次后会被弹飞。为了使得游戏更有趣,Lostmonkey可以修改某个弹力装置的弹力系数,任何时候弹力系数均为正整数。
输入:
第一行包含一个整数n,表示地上有n个装置,装置的编号从0到n-1,接下来一行有n个正整数,依次为那n个装置的初始弹力系数。第三行有一个正整数m,接下来m行每行至少有两个数i、j,若i=1,你要输出从j出发被弹几次后被弹飞,若i=2则还会再输入一个正整数k,表示第j个弹力装置的系数被修改成k。对于20%的数据n,m<=10000,对于100%的数据n<=200000,m<=100000
输出:
对于每个i=1的情况,你都要输出一个需要的步数,占一行。
样例输入:
4
1 2 1 1
3
1 1
2 1 1
1 1
样例输出:
2
3
题解:
把每个位置看成一个点,每个点往后跳的位置看做是树的边,新建一个点表示被弹飞的点(根节点)。你需要支持修改边,查询某个点的深度。lct果题。(第一次写lct)
代码:
#include <cstdio> #include <cstring> #include <algorithm> #include <cmath> #ifdef WIN32 #define LL "%I64d" #else #define LL "%lld" #endif #ifdef CT #define debug(...) printf(__VA_ARGS__) #define setfile() #else #define debug(...) #define filename "" #define setfile() freopen(filename".in", "r", stdin); freopen(filename".out", "w", stdout); #endif #define R register #define getc() (S == T && (T = (S = B) + fread(B, 1, 1 << 15, stdin), S == T) ? EOF : *S++) #define dmax(_a, _b) ((_a) > (_b) ? (_a) : (_b)) #define dmin(_a, _b) ((_a) < (_b) ? (_a) : (_b)) #define cmax(_a, _b) (_a < (_b) ? _a = (_b) : 0) #define cmin(_a, _b) (_a > (_b) ? _a = (_b) : 0) char B[1 << 15], *S = B, *T = B; inline int FastIn() { R char ch; R int cnt = 0; R bool minus = 0; while (ch = getc(), (ch < '0' || ch > '9') && ch != '-') ; ch == '-' ? minus = 1 : cnt = ch - '0'; while (ch = getc(), ch >= '0' && ch <= '9') cnt = cnt * 10 + ch - '0'; return minus ? -cnt : cnt; } #define maxn 200010 struct Node *null; struct Node { Node *ch[2], *fa; int size; inline void update() { size = ch[0] -> size + ch[1] -> size + 1; } inline bool type() { return fa -> ch[1] == this; } inline bool check() { return fa -> ch[type()] == this; } inline void rotate() { R Node *f = fa; R bool d = type(); (fa = f -> fa), f -> check() ? fa -> ch[f -> type()] = this : 0; (f -> ch[d] = ch[!d]) != null ? ch[!d] -> fa = f : 0; f -> fa = this; ch[!d] = f; f -> update(); } inline void splay() { for (; check(); rotate()) if (fa -> check()) (type() != fa -> type() ? this : fa) -> rotate(); update(); } inline Node *access() { R Node *i = this, *j = null; for (; i != null; i = (j = i) -> fa) { i -> splay(); i -> ch[1] = j; i -> update(); } return j; } inline void make_root() { access(); splay(); } inline void link(R Node *that) { make_root(); ch[0] -> fa = null; ch[0] = null; fa = that; update(); } }mem[maxn], *tot = mem, *fir = mem; int main() { // setfile(); null = mem; null -> size = 0; null -> fa = null -> ch[0] = null -> ch[1] = null; R int n = FastIn(); for (R int i = 1; i <= n; ++i) *++tot = (Node) {null, null, null, 1}; for (R int i = 1; i <= n; ++i) { R int x = FastIn(); if (i + x <= n) (mem + i) -> fa = mem + i + x; } R int m = FastIn(); for (R int i = 1; i <= m; ++i) { R int opt = FastIn(), x = FastIn() + 1; if (opt == 1) { R Node *now = fir + x; now -> make_root(); printf("%d\n",now -> size ); } else { R int k = FastIn(); R Node *now = fir + x; now -> link(x + k > n ? null : now + k); } } return 0; } /* 4 1 2 1 1 3 1 1 2 1 1 1 1 */