传送门:洛谷:[HNOI2010]弹飞绵羊
某天,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。
对于每个i=1的情况,你都要输出一个需要的步数,占一行。
对于20%的数据n,m<=10000,对于100%的数据n<=200000,m<=100000
其实很好理解,分成 n−−√ n 个块,处理从前一个块到下一个块的个数和到下一个块的那个点,网上讲解很多。
先自己打了个非常丑的代码,居然得了30分(不得不吐槽洛谷的水数据)。
然后看了黄学长的代码,emmm,非常妙。
明天补LCT做法。
然而两天后。LCT见篇尾。
我的30分代码:
#include
#include
#include
#include
#include
#include
using namespace std;
const int N=2e5+10;
int n,m,t[N],siz,mx,tot,lc[N];
bool vis[N];
vector<int>ss[500];
struct Node{
int to;
int c;
}st[N];
inline int read()
{
char c=getchar();int x=0,t=1;
while(c<'0' || c>'9') {if(c=='-') t=-1;c=getchar();}
while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
return x*t;
}
inline void add()
{
for(int i=1;i<=n;i++){
int k=i/mx;
if(!vis[i]){
if(k%mx==0 && k!=0) ss[k-1].push_back(i);
else ss[k].push_back(i);
vis[i]=true;
lc[i]=i;
st[i].c=1;
int now=i+t[i];
int www=i%mx==0? i:(k+1)*mx;
for(;now<=www && now<=n;now+=t[now]){
vis[now]=true;lc[now]=i;st[i].c++;
}
if(now<=n){
st[i].to=now;
}
}
}
}
inline int query(int k)
{
int ans=1;
int w=lc[k];
while(w!=k){
w+=t[w];ans++;
}
ans=st[lc[k]].c-ans;
w=st[lc[k]].to;
while(w!=0){
int pw=lc[w],aa=0;
while(pw!=w){
pw+=t[pw];aa++;
}
ans+=st[lc[w]].c-aa;
w=st[w].to;
}
return ans;
}
inline void change(int num,int v)
{
int k=num/mx,o=num,ti=0;t[num]=v;
int www= num%mx==0? num:(k+1)*mx;
for(;o<=www && o<=n;o+=t[o]){
ti++;
}
if(lc[num]==num){
st[num].to=o>n? 0:o;
st[num].c=ti;
}
k-=num%mx==0 && k!=0?1:0;
int len=ss[k].size();
for(int i=0;iint temp=ss[k][i],ptr=0;
int nn=temp;
while(nn!=st[temp].to){
ptr++;
nn+=t[nn];
if(nn==num){
st[temp].to=o>n? 0:o;
st[temp].c=ptr+ti;
break;
}
}
}
}
int main(){
memset(vis,false,sizeof(vis));
n=read();mx=(int)(sqrt(n));
for(int i=1;i<=n;i++){
t[i]=read();
}
add();m=read();
while(m--){
int op=read();
if(op==1){
printf("%d\n",query(read()));
}else{
int num=read()+1,val=read();
change(num,val);
}
}
//system("pause");
return 0;
}
//anyway luogu->30 st->0
AC代码:
分块
#include
#include
#include
#include
#include
using namespace std;
const int N=2e5+10;
int n,t[N],m,mx;
int st[N],ed[N],b[N];
int l[500];
inline int read()
{
char c=getchar();int x=0,t=1;
while(c<'0' || c>'9') {if(c=='-') t=-1;c=getchar();}
while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
return x*t;
}
inline int query(int k)
{
int ans=0;
while(k!=0){
ans+=st[k];
k=ed[k];
}
return ans;
}
inline void add(int k,int v)
{
t[k]=v;
for(int i=k;i>=l[b[k]];i--){
st[i]=1;
if(i+t[i]>n) {
ed[i]=0;continue;
}
if(b[i]==b[i+t[i]]){
st[i]+=st[i+t[i]];
ed[i]=ed[i+t[i]];
}else{
ed[i]=i+t[i];
}
}
}
int main(){
n=read();mx=(int)(sqrt(n));
for(int i=1;i<=n;i++) t[i]=read();
int cnt=n/mx;
if(n%mx!=0) cnt++;
for(int i=0;i1;
for(int i=n;i>=1;i--){
b[i]=(i-1)/mx;st[i]=1;
if(i+t[i]>n) continue;
else if(b[i]==b[i+t[i]]){
st[i]+=st[i+t[i]];ed[i]=ed[i+t[i]];
}else ed[i]=i+t[i];
}
m=read();
while(m--){
int op=read();
if(op==1){
printf("%d\n",query(read()+1));
//query(read())-1 -> query(read()+1)
//WA->AC
}else{
int p=read()+1,q=read();
//p->题意上的问题
add(p,q);
}
}
return 0;
}
LCT
#include
#include
#include
#include
#include
using namespace std;
const int N=2e5+10;
int n,m,to[N],t[N];
int f[N],rev[N],c[N][2],siz[N];
int now[N];
inline int read(){
char ch=getchar();int x=0;
while(ch<'0' || ch>'9') ch=getchar();
while(ch>='0' && ch<='9') {x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
return x;
}
inline bool isroot(int x)
{
return (c[f[x]][0]!=x && c[f[x]][1]!=x);
}
inline void pushup(int k)
{
siz[k]=siz[c[k][0]]+siz[c[k][1]]+1;
}
inline void pushdown(int k)
{
if(rev[k]){
rev[c[k][0]]^=1;
rev[c[k][1]]^=1;
rev[k]=0;
swap(c[k][0],c[k][1]);
}
}
inline void rotate(int x)
{
int y=f[x];int z=f[y];
if(!isroot(y)){
c[z][c[z][1]==y]=x;
}
int ptr= c[y][1]==x;
f[x]=z;f[y]=x;f[c[x][ptr^1]]=y;
c[y][ptr]=c[x][ptr^1];c[x][ptr^1]=y;
pushup(y);pushup(x);
}
inline void splay(int x)
{
int tot=0,p=x;now[++tot]=x;
for(;!isroot(p);p=f[p]) now[++tot]=f[p];
for(int i=tot;i>0;i--) pushdown(now[i]);
while(!isroot(x)){
int y=f[x];int z=f[y];
if(!isroot(y)){
if(c[z][0]==y ^ c[y][0]==x) rotate(x);
else rotate(y);
}
rotate(x);
}
}
inline void access(int x)
{
int t=0;
while(x){
splay(x);
c[x][1]=t;
t=x;x=f[x];
}
}
inline void rever(int x)
{
access(x);splay(x);rev[x]^=1;
}
inline int query(int x)
{
rever(n+1);access(x);splay(x);
return siz[c[x][0]];
}
inline void cut(int x,int y)
{
rever(x);access(y);splay(y);c[y][0]=f[x]=0;
}
inline void link(int x,int y)
{
rever(x);f[x]=y;splay(x);
}
inline void update(int x,int val)
{
int p=min(n+1,x+val);
cut(x,to[x]);link(x,p);
to[x]=p;
}
int main(){
n=read();
for(int i=1;i<=n;i++) {
t[i]=read();
to[i]= (i+t[i])>n+1? n+1:i+t[i];
f[i]=to[i];
}
//for(int i=1;i<=n;i++) siz[i]=1;
m=read();
while(m--){
int op=read();
if(op==1){
printf("%d\n",query(read()+1));
}else{
int x=read()+1,rp=read();
update(x,rp);
}
}
return 0;
}
//读错题了orzz
//怪不得一直想不通为什么要read()+1
//是从编号为read()的开始
//laji题面