您的位置:首页 > 其它

[Codeforces]856D - Masha and Cactus

2017-09-12 13:20 274 查看
题目大意:给出一棵树和若干条可以加入的边,要求加入若干条边使图是仙人掌并且加入的边权和最大,仙人掌定义为没有一个点属于超过1个环。(n,m<=200,000)

做法:这题的仙人掌跟平时见到的不太一样,是以点为判定条件,比较坑……发现与选若干条不相交的链权值和最大等价,dp,f[i]表示i的子树内最大权值和,每次枚举一条lca为i的链转移,需要求出这条链以外的子树的dp值,树剖后线段树维护每个点轻儿子dp值之和,重儿子的暴力算一算就可以了,复杂度O((n+m)log^2)。

代码:

#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
inline int read()
{
int x;char c;
while((c=getchar())<'0'||c>'9');
for(x=c-'0';(c=getchar())>='0'&&c<='9';)x=x*10+c-'0';
return x;
}
#define MN 200000
#define N 262144
#define mp(x,y) make_pair(x,y)
struct edge{int nx,t;}e[MN+5];
int h[MN+5],en,s[MN+5],mx[MN+5],d[MN+5],fa[MN+5],f[MN+5],l[MN+5],p[MN+5],cnt,t[N*2+5],F[MN+5];
vector<pair<pair<int,int>,int> > v[MN+5];
inline void ins(int x,int y){e[++en]=(edge){h[x],y};h[x]=en;}
void add(int k,int x){for(k+=N;k;k>>=1)t[k]+=x;}
int query(int l,int r)
{
int res=0;
if(l<=r)for(l+=N-1,r+=N+1;l^r^1;l>>=1,r>>=1)
{
if(~l&1)res+=t[l+1];
if( r&1)res+=t[r-1];
}
return res;
}
void dfs(int x)
{
s[x]=1;
for(int i=h[x];i;i=e[i].nx)
{
fa[e[i].t]=x;d[e[i].t]=d[x]+1;
dfs(e[i].t);
s[x]+=s[e[i].t];
if(s[e[i].t]>s[mx[x]])mx[x]=e[i].t;
}
}
void build(int x)
{
p[l[x]=++cnt]=x;
if(mx[x])f[mx[x]]=f[x],build(mx[x]);
for(int i=h[x];i;i=e[i].nx)if(e[i].t!=mx[x])build(f[e[i].t]=e[i].t);
}
int lca(int x,int y)
{
while(f[x]!=f[y])if(d[f[x]]>d[f[y]])x=fa[f[x]];else y=fa[f[y]];
return d[x]<d[y]?x:y;
}
int up(int x,int y)
{
for(;f[x]!=f[y];x=fa[f[x]])
if(fa[f[x]]==y)return f[x];
return p[l[y]+1];
}
int Q(int x,int y)
{
int res=0;
for(;f[x]!=f[y];x=fa[f[x]])
res+=query(l[f[x]],l[x])+F[mx[x]]-F[f[x]];
return res+query(l[y],l[x])+F[mx[x]];
}
void dp(int x)
{
for(int i=h[x];i;i=e[i].nx)dp(e[i].t),F[x]+=F[e[i].t];
add(l[x],F[x]-F[mx[x]]);
for(int i=0;i<v[x].size();++i)
{
int y=up(v[x][i].first.first,x);
F[x]=max(F[x],Q(v[x][i].first.first,y)+Q(v[x][i].first.second,x)-F[y]+v[x][i].second);
}
}
int main()
{
int n,m,i,x,y;
n=read();m=read();
for(i=2;i<=n;++i)ins(read(),i);
dfs(1);build(f[1]=1);
while(m--)
{
i=lca(x=read(),y=read());
if(x==i)swap(x,y);
v[i].push_back(mp(mp(x,y),read()));
}
dp(1);
printf("%d",F[1]);
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: