题意:
一条直线有 n n n 个洞
每当一个球掉进洞中时,回弹到后面距它 l i l_i li 的洞中或者直接被弹飞,要求支持两种操作:
-
修改某个洞的弹力 l i l_i li
-
询问小球进入某个洞后,被弹几次之后会飞出,飞出之前进入的最后一个洞是什么。
将序列分成 n \sqrt n n 个块,对于每个位置 i i i 我们记录从每个位置开始跳出这个块需要的时间 t i t_i ti 和跳到的位置 n x t i nxt_i nxti
对于每组询问,按照记录的信息跳,时间复杂度 O ( n ) O(\sqrt n) O(n)
对于修改,我们只需要修改和 i i i 在同一块里的所有位置信息,按照从右往左的顺序,信息 O ( 1 ) O(1) O(1) 可以处理,那么时间复杂度依旧是 O ( n ) O(\sqrt n) O(n)
总时间复杂度 O ( n n ) O(n\sqrt n) O(nn)
#include <map>
#include <set>
#include <ctime>
#include <queue>
#include <stack>
#include <cmath>
#include <vector>
#include <bitset>
#include <cstdio>
#include <cctype>
#include <string>
#include <numeric>
#include <cstring>
#include <cassert>
#include <climits>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std ;
//#define int long long
#define rep(i, a, b) for (int i = (a); i <= (b); i++)
#define per(i, a, b) for (int i = (a); i >= (b); i--)
#define loop(s, v, it) for (s::iterator it = v.begin(); it != v.end(); it++)
#define cont(i, x) for (int i = head[x]; i; i = e[i].nxt)
#define clr(a) memset(a, 0, sizeof(a))
#define ass(a, sum) memset(a, sum, sizeof(a))
#define lowbit(x) (x & -x)
#define all(x) x.begin(), x.end()
#define ub upper_bound
#define lb lower_bound
#define pq priority_queue
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define iv inline void
#define enter cout << endl
#define siz(x) ((int)x.size())
#define file(x) freopen(#x".in", "r", stdin),freopen(#x".out", "w", stdout)
typedef long long ll ;
typedef unsigned long long ull ;
typedef pair <int, int> pii ;
typedef vector <int> vi ;
typedef vector <pii> vii ;
typedef queue <int> qi ;
typedef queue <pii> qii ;
typedef set <int> si ;
typedef map <int, int> mii ;
typedef map <string, int> msi ;
const int N = 100010 ;
const int M = 320 ;
const int INF = 0x3f3f3f3f ;
const int iinf = 1 << 30 ;
const ll linf = 2e18 ;
const int MOD = 1000000007 ;
const double eps = 1e-7 ;
void print(int x) { cout << x << endl ; exit(0) ; }
void PRINT(string x) { cout << x << endl ; exit(0) ; }
void douout(double x){ printf("%lf\n", x + 0.0000000001) ; }
int a[N], bl[N], l[M], r[M], t[N], nxt[N], la[N] ;
int n, m, len, num ;
void build() {
len = sqrt(n), num = (n - 1) / len + 1 ;
rep(i, 1, n) bl[i] = (i - 1) / len + 1 ;
rep(i, 1, num) l[i] = (i - 1) * len + 1, r[i] = i * len ; r[num] = n ;
}
void modify(int x) {
per(i, r[x], l[x]) {
if (i + a[i] > r[x]) {
t[i] = 1 ;
la[i] = i ;
nxt[i] = i + a[i] ;
} else {
t[i] = t[i + a[i]] + 1 ;
la[i] = la[i + a[i]] ;
nxt[i] = nxt[i + a[i]] ;
}
}
}
void query(int x) {
int tim = 0, lst = x ;
while (x <= n) {
tim += t[x] ;
lst = la[x] ;
x = nxt[x] ;
}
printf("%d %d\n", lst, tim) ;
}
signed main(){
scanf("%d%d", &n, &m) ;
rep(i, 1, n) scanf("%d", &a[i]) ;
build() ;
rep(i, 1, num) modify(i) ;
while (m--) {
int op, x, y ; scanf("%d", &op) ;
if (op == 0) {
scanf("%d%d", &x, &y) ;
a[x] = y ;
modify(bl[x]) ;
} else {
scanf("%d", &x) ;
query(x) ;
}
}
return 0 ;
}
/*
写代码时请注意:
1.ll?数组大小,边界?数据范围?
2.精度?
3.特判?
4.至少做一些
思考提醒:
1.最大值最小->二分?
2.可以贪心么?不行dp可以么
3.可以优化么
4.维护区间用什么数据结构?
5.统计方案是用dp?模了么?
6.逆向思维?
*/