题解:P10238 [yLCPC2024] F. PANDORA PARADOXXX
离线倒着做,变成连边,以及询问所有连通块直径最大值。
然后是一个经典结论,就是当边权非负时,两个连通块合并,新的直径端点必然是之前两个连通块的四个端点的其中两个,于是你可以使用并查集维护连通块,以及每个连通块的直径端点。合并的时候大力分讨几种即可。
证明:
考虑新的直径,要么在原来的两个连通块内部,要么横跨连通块。在连通块内部的时候,两个端点就是之前两部分的端点。否则,由于树中离每个点最远的点必然在直径端点上,这个就必然成立了。
#include <bits/stdc++.h>
using namespace std;
//#define int long long
using ll = long long;
const int N = 1e5 + 5, MOD = 1e9 + 7, HSMOD = 1610612741, HSMOD2 = 998244353; // Remember to change
int n, m, q, t, a[N];
mt19937 rnd(chrono::system_clock::now().time_since_epoch().count());
long long qpow(long long a, long long b)
{
long long res = 1ll, base = a;
while (b)
{
if (b & 1ll) res = res * base % MOD;
base = base * base % MOD;
b >>= 1ll;
}
return res;
}
bool isprime(int x)
{
if (x == 1) return 0;
for (int i = 2; 1ll * i * i <= x; i++) if (x % i == 0) return 0;
return 1;
}
namespace FastIo
{
#define QUICKCIN ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
int read()
{
char ch = getchar();
int x = 0, f = 1;
while ((ch < '0' || ch > '9') && ch != '-') ch = getchar();
while (ch == '-')
{
f = -f;
ch = getchar();
}
while (ch >= '0' && ch <= '9')
{
x = (x << 1) + (x << 3) + (ch ^ 48);
ch = getchar();
}
return x * f;
}
template<class T>
void write(T x)
{
if (x < 0)
{
putchar('-');
x = -x;
}
if (x > 9) write(x / 10);
putchar(x % 10 + '0');
}
template<class T>
void writeln(T x)
{
write(x);
putchar('\n');
}
}
template<typename T>
class Bit
{
public:
T lowbit(T x)
{
return x & -x;
}
T tr[N];
void add(T x, T y)
{
while (x < N)
{
tr[x] += y;
x += lowbit(x);
}
}
T query(T x)
{
T sum = 0;
while (x)
{
sum += tr[x];
x -= lowbit(x);
}
return sum;
}
};
int u[N],v[N];
ll w[N];
vector<pair<int,ll>> G[N];
int fa[N][21];
ll dis[N];
int dep[N];
void dfs(int u,int f,int d)
{
fa[u][0]=f;
dep[u]=d;
for(auto&[j,w]:G[u])
{
if(j!=f)
{
dis[j]=dis[u]+w;
dfs(j,u,d+1);
}
}
}
ll res[N];
inline int LCA(int u,int v)
{
if(u==v) return u;
if(dep[u]<dep[v]) swap(u,v);
int k=dep[u]-dep[v],c=0;
while(k)
{
if(k&1) u=fa[u][c];
c++;
k>>=1;
}
if(u==v) return u;
for(int i=20;i>=0;i--) if(fa[u][i]^fa[v][i]) u=fa[u][i],v=fa[v][i];
return fa[u][0];
}
inline ll getdis(int u,int v)
{
return dis[u]+dis[v]-2*dis[LCA(u,v)];
}
ll ans=-1ll;
class Union_Find
{
public:
int fa[N];
pair<int,int> diamt[N];
void Init()
{
for(int i=1;i<=n;i++) fa[i]=i,diamt[i]=make_pair(i,i);
}
int find(int u)
{
return (fa[u]==u?u:fa[u]=find(fa[u]));
}
void merge(int u,int v)
{
if((u=find(u))==(v=find(v))) return;
vector<int> ver;
ver.emplace_back(diamt[u].first);
ver.emplace_back(diamt[u].second);
ver.emplace_back(diamt[v].first);
ver.emplace_back(diamt[v].second);
int ru=0,rv=0;
ll md=-1ll;
for(int i=0;i<4;i++)
{
for(int j=i+1;j<4;j++)
{
ll nd=getdis(ver[i],ver[j]);
if(nd>md) md=nd,ru=ver[i],rv=ver[j];
}
}
ans=max(ans, md);
fa[v]=u;
diamt[u]=make_pair(ru,rv);
}
}dsu;
bool flag[N];
int main()
{
ios::sync_with_stdio(0), cin.tie(0);
cin>>t;
while(t--)
{
cin>>n>>q;
dsu.Init();
for(int i=1;i<=n;i++) G[i].clear(),G[i].shrink_to_fit(),flag[i]=0;
for(int i=1;i<n;i++)
{
cin>>u[i]>>v[i]>>w[i];
G[u[i]].emplace_back(make_pair(v[i],w[i]));
G[v[i]].emplace_back(make_pair(u[i],w[i]));
}
dfs(1,0,1);
vector<int> del;
for(int i=1;i<=q;i++)
{
int e;
cin>>e;
flag[e]=1;
del.emplace_back(e);
}
ans=0ll;
reverse(del.begin(),del.end());
for(int j=1;j<=20;j++) for(int i=1;i<=n;i++) fa[i][j]=fa[fa[i][j-1]][j-1];
for(int i=1;i<n;i++) if(!flag[i]) dsu.merge(u[i],v[i]);
int idx=q;
for(auto&j:del)
{
res[idx]=ans;
idx--;
dsu.merge(u[j],v[j]);
}
for(int i=1;i<=q;i++) cout<<res[i]<<"\n";
}
return 0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现