題目鏈接：點擊查看
題目大意：給出一棵有 n 個節點的樹，需要執行 m 次操作：
1 x：點 x 被染色，且隨后每一秒染色的區域都會向外擴展一個單位2 x：清空所有的染色3 x：問點 x 是否被染色
題目分析：因為可以離線，所以考慮按照詢問分塊，首先用序列自動機求出每次操作后的下一次操作 2，然后分兩種情況處理：
如果兩次操作 2 的間隔大于 sqrt( m )，直接暴力 bfs 去模擬，每一塊的時間復雜度為 O( n )，均攤之后是 O( sqrt( m ) * n )如果兩次操作 2 的間隔小于 sqrt( m )，直接 O( m?) 去枚舉所有操作 1 ，再 O( m ) 枚舉所有操作 3 ，分別計算貢獻，均攤之后是 O( sqrt( m ) * sqrt( m ) * sqrt( m ) ) = O( sqrt( m ) * m )
所以總時間復雜度為 O( sqrt( m ) * ( n + m ) )
間隔大于 sqrt( m ) 時直接按照題意 bfs 就行了，這個沒有什么技術含量，重點說一下如何處理間隔小于 sqrt( m ) 的操作，對于一個操作 1 和一個操作 3 來說，如果操作 1 會影響到操作 3 的話，那么其在樹上的距離一定是小于操作之間的間隔的，這樣就轉換成距離問題了，預處理一下深度，每次求一下 LCA 就能快速計算了
但是如果是普通的樹上倍增或者樹鏈剖分來求 LCA 的話，還要帶一個 log，或許也可以通過本題，但最優的方法是用 tarjan 離線 O( n ) 處理出所需要的 lca 然后進行處理，或者用 RMQ 預處理一下，同樣可以 O( 1 ) 查詢 LCA
代碼：
?
//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=1e5+100;vector<int>node[N];int op[N],x[N],nt[N],ans[N];bool vis[N];
/*RMQ求LCA*/
int dfn[N],id[N<<1][25],deep[N],cnt,lg2[N<<1];void dfs(int u,int fa,int dep)
{deep[u]=dep;dfn[u]=++cnt;id[cnt][0]=u;for(auto v:node[u]){if(v==fa)continue;dfs(v,u,dep+1);id[++cnt][0]=u;}
}void RMQ()
{for(int i=2;i<=cnt;i++)lg2[i]=lg2[i>>1]+1;for(int j=1;j<=20;j++)for(int i=1;(i+(1<<j)-1)<=cnt;i++){int l=i,r=i+(1<<(j-1));id[i][j]=deep[id[l][j-1]]<deep[id[r][j-1]]?id[l][j-1]:id[r][j-1];}
}int LCA(int x,int y)
{int l=dfn[x],r=dfn[y];if(l>r)swap(l,r);int k=lg2[r-l+1];return deep[id[l][k]]<deep[id[r-(1<<k)+1][k]]?id[l][k]:id[r-(1<<k)+1][k];
}
/*RMQ求LCA*/
int get_dis(int x,int y)
{return deep[x]+deep[y]-2*deep[LCA(x,y)];
}void solve(int l,int r)
{queue<int>q;memset(vis,false,sizeof(vis));for(int i=l;i<=r;i++){int temp=q.size();while(temp--){int u=q.front();q.pop();for(auto v:node[u]){if(vis[v])continue;vis[v]=true;q.push(v);}}if(op[i]==1&&!vis[x[i]]){q.push(x[i]);vis[x[i]]=true;}if(op[i]==3){if(vis[x[i]])ans[i]=true;}}
}int main()
{
#ifndef ONLINE_JUDGE
// freopen("data.in.txt","r",stdin);
// freopen("data.out.txt","w",stdout);
#endif
// ios::sync_with_stdio(false);int n,m;scanf("%d%d",&n,&m);for(int i=1;i<n;i++){int u,v;scanf("%d%d",&u,&v);node[u].push_back(v);node[v].push_back(u);}dfs(1,0,0);RMQ();for(int i=1;i<=m;i++)scanf("%d%d",op+i,x+i);nt[m+1]=m+1;for(int i=m;i>=1;i--)//nt[i]:記錄第i個位置后再次被重置的位置 {nt[i]=nt[i+1];if(op[i]==2)nt[i]=i;}int b=sqrt(m);for(int i=1;i<=m;i++){if(nt[i]-i>b){solve(i,nt[i]-1);}else{for(int j=i;j<nt[i];j++)//處理每個op==1對每個op==3的貢獻 if(op[j]==1)for(int k=j+1;k<nt[i];k++)if(op[k]==3)if(get_dis(x[j],x[k])<=k-j)ans[k]=true; }i=nt[i];}for(int i=1;i<=m;i++)if(op[i]==3)printf("%s\n",ans[i]?"wrxcsd":"orzFsYo");return 0;
}
?
總結
以上是生活随笔為你收集整理的牛客 - 牛半仙的妹子Tree(按询问分块+RMQ求LCA)的全部內容，希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。
