CF1942F Farmer John's Favorite Function

题意简述

定义 \(f_i=\sqrt{f_{i-1}+a_i},f_0=0\)。

有 \(q\) 次操作，每次操作单点修改一个 \(a_i\) 的值，每次修改后求 \(\lfloor f_n\rfloor\) 的值。

\(n,q\le 2\times10^5,0\le a_i\le 10^{18}\)。

分析

发现这个 \(f_i\) 不是整数非常不利于我们做题，考虑将它变成整数。

令新的 \(f'_i=\lfloor f_i\rfloor\)，这样的话 \(f'_{i+1}=\lfloor\sqrt{f'_i+a_{i+1}}\rfloor=\lfloor\sqrt{f_i-x+a_{i+1}}\rfloor(0\le x<1)\)，但是我们发现这个 \(x\) 不会影响 \(f'_{i+1}=\lfloor\sqrt{f'_i+a_{i+1}}\rfloor\) 的取值，因为 \(f'_i+a_{i+1}\) 是整数，加上了 \(x\) 也只是改变了小数部分，对整数部分没有影响，所以有 \(\lfloor\sqrt{f'_i+a_{i+1}}\rfloor=\lfloor\sqrt{f_i+a_{i+1}}\rfloor\)，故 \(f'_i=f_i\)，\(f_i\) 就转化成整数了。

直觉发现根号这个东西数字缩小的非常快，通过手玩计算器可以发现 \(10^{18}\) 在取六次根号后下取整的值就变成了 \(1\)，由此当 \(n\ge 6\) 时，\(a_1\) 的改变只会使 \(f_n\) 的值至多变化 \(1\)。保险起见我把这个阈值开到了 \(10\)。

考虑如何处理单点修改。将原序列分块，设一个块维护的区间为 \([l,r]\)，块内维护 \(f_i\) 表示当 \(f_{l-1}=0\) 时，\(f_r\) 的取值；维护 \(g_i\) 表示当 \(f_{l-1}\ge g_i\) 时 \(f_i\) 的值将会增长 \(1\)（根据第二个结论，\(f_i\) 的值只会至多增长 \(1\)）。

\(f_i\) 的值直接递推就好，求解 \(g_i\) 考虑逆推，假设最终的 \(f'_r=f_r+1\)，\(f'_{r-1}\) 的值可以由 \(f_r^{'2}-a_r\) 求出，以此类推直到推出 \(f'_{l-1}\) 的值，求出的 \(f'_{l-1}\) 就是 \(g_i\) 的值了。当然你可以二分答案以多一个 log 的代价求出。

每次修改暴力计算块内信息即可，查询的话直接递推，设计算前 \(i-1\) 个块后的答案为 \(ans\)，若 \(ans\ge g_i\)，则 \(ans=f_i+1\)，否则 \(ans=f_i\)。

时间复杂度 \(O(n\sqrt n)\)，视 \(n,q\) 同阶。

注意分块时有可能块长小于阈值（此时第二个结论不成立），对于这种块暴力计算即可。

点击查看代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<map>
#include<unordered_map>
#include<vector>
#include<queue>
#include<bitset>
#include<set>
#include<ctime>
#include<random>
#define x1 xx1
#define y1 yy1
#define IOS ios::sync_with_stdio(false)
#define ITIE cin.tie(0);
#define OTIE cout.tie(0);
#define PY puts("Yes")
#define PN puts("No")
#define PW puts("-1")
#define P__ puts("")
#define PU puts("--------------------")
#define popc __builtin_popcount
#define mp make_pair
#define fi first
#define se second
#define gc getchar
#define pc putchar
#define pb emplace_back
#define rep(a,b,c) for(int a=(b);a<=(c);++a)
#define per(a,b,c) for(int a=(b);a>=(c);--a)
#define reprange(a,b,c,d) for(int a=(b);a<=(c);a+=(d))
#define perrange(a,b,c,d) for(int a=(b);a>=(c);a-=(d))
#define graph(i,j,k,l) for(int i=k[j];i;i=l[i].nxt)
#define lowbit(x) (x&-x)
#define lson(x) (x<<1)
#define rson(x) (x<<1|1)
#define mem(x,y) memset(x,y,sizeof x)
//#define double long double
#define int long long
//#define int __int128
using namespace std;
typedef long long i64;
using pii=pair<int,int>;
bool greating(int x,int y){return x>y;}
bool greatingll(long long x,long long y){return x>y;}
inline int rd(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}return x*f;
}
inline void write(int x,char ch='\0'){
	if(x<0){x=-x;putchar('-');}
	int y=0;char z[40];
	while(x||!y){z[y++]=x%10+48;x/=10;}
	while(y--)putchar(z[y]);if(ch!='\0')putchar(ch);
}
bool Mbg;
const int maxn=2e5+5,maxm=805,inf=0x3f3f3f3f,B=300,BMIN=10;
const long long llinf=0x3f3f3f3f3f3f3f3f;
int n,Q,num,a[maxn];
int L[maxm],R[maxm],bel[maxn];
int f[maxm],g[maxm];
void upd(int x){
	int res=0;
	rep(i,L[x],R[x])res=floor(sqrtl(res+a[i]));
	f[x]=res;
	res=f[x]+1;
	per(i,R[x],L[x]){
		int _res=res*res-a[i];
		if(_res>inf)return g[x]=llinf,void();
		res=_res;
	}
	g[x]=res;
}
int qry(){
	int res=0;
	rep(i,1,num){
		if(R[i]-L[i]+1<=BMIN){
			rep(j,L[i],R[i])res=floor(sqrtl(res+a[j]));
		}else{
			if(res>=g[i])res=f[i]+1;
			else res=f[i];
		}
	}
	return res;
}
void solve_the_problem(){
	n=rd(),Q=rd(),num=(n+B-1)/B;rep(i,1,n)a[i]=rd();
	rep(i,1,num){
		L[i]=R[i-1]+1,R[i]=min(n,R[i-1]+B);
		rep(j,L[i],R[i])bel[j]=i;
		upd(i);
	}
	while(Q--){
		int x=rd(),y=rd();
		a[x]=y,upd(bel[x]);
		write(qry(),10);
	}
}
bool Med;
signed main(){
//	freopen(".in","r",stdin);freopen(".out","w",stdout);
//	fprintf(stderr,"%.3lfMB\n",(&Mbg-&Med)/1048576.0);
	int _=1;while(_--)solve_the_problem();
}
/*

*/