CF1389G Directing Edges

一、题目

点此看题

二、解法

这个图很不舒服，如果想把它转化成舒服的结构-----树，那么我们就要考虑缩点。

这个图的性质能不能缩点呢？怎么缩点呢？考虑缩成边双联通分量，我们就可以把这个部分设置成互相可到达的有向图，利用的性质就是边双任意两点之间有两条经过边完全不同的路径。

转化成树之后，对于这种规划问题我们考虑树形$dp$，设$f[i]$为$i$子树内所有所有特殊点到$i$的最大收益，如果我们把某一条边加上，那么可以获得$f[v]$的贡献，初始可以获得当前点上的收益$a[u]$：
$$f[u]=\max (f[v]-w,0)+a[u]$$还有一点小问题，因为这个转移需要保证子树内有关键点，如果没有的话怎么办？我们把这一点的权值缩到最近是特殊点的祖先上，这些点一定是一条船上的蚂蚱！

但是我们要求每一个点为根的答案，没关系，再跑一遍树形$dp$，设$g[i]$为$i$是根时候的收益，这是候我们只需要多考虑他父亲方向的贡献就行了：
$$g[u]=f[u]+\max (g[fa]-\max (f[u]-w,0)-w,0)$$时间复杂度$O(n)$

#include <cstdio>
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
const int M = 300005;
#define int long long
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,m,k,tot,rt,Index,cnt,f[M],dfn[M],low[M],col[M];
int w[M],fa[M],dp[M][2],a[M],c[M],sa[M],sc[M],suma[M];
stack<int> s;
struct edge
{
	int v,c,next;
}e[2*M];vector<edge> g[M];
void tarjan(int u,int p)
{
	dfn[u]=low[u]=++Index;
	s.push(u);
	for(int i=f[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==p) continue;
		if(!dfn[v])
		{
			tarjan(v,u);
			low[u]=min(low[u],low[v]);
		}
		else
			low[u]=min(low[u],dfn[v]);
	}
	if(low[u]==dfn[u])
	{
		int v;cnt++;
		do
		{
			v=s.top();s.pop();
			col[v]=cnt;
			sa[cnt]|=a[v];
			sc[cnt]+=c[v];
		}while(u!=v);
		if(sa[cnt]) rt=cnt;
	}
}
void dfs1(int u,int p)
{
	suma[u]=sa[u];
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i].v;
		if(v==p) continue;
		dfs1(v,u);
		suma[u]|=suma[v];
		if(suma[v]==0)
		{
			fa[v]=u;
			sc[u]+=sc[v];
		}
	}
}
void dfs2(int u,int p)
{
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i].v,c=g[u][i].c;
		if(v==p || suma[v]==0) continue;
		dfs2(v,u);
		dp[u][0]+=max(dp[v][0]-c,0ll);
	}
	dp[u][0]+=sc[u];
}
void dfs3(int u,int p)
{
	dp[u][1]+=dp[u][0];
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i].v,c=g[u][i].c;
		if(v==p || suma[v]==0) continue;
		dp[v][1]+=max(dp[u][1]-max(dp[v][0]-c,0ll)-c,0ll);
		dfs3(v,u);
	}
}
int find(int x)
{
	if(x!=fa[x]) fa[x]=find(fa[x]);
	return fa[x];
}
signed main()
{
	n=read();m=read();k=read();
	for(int i=1;i<=k;i++)
		a[read()]=1;
	for(int i=1;i<=n;i++)
		c[i]=read(),fa[i]=i;
	for(int i=1;i<=m;i++)
		w[i]=read();
	for(int i=1;i<=m;i++)
	{
		int u=read(),v=read();
		e[++tot]=edge{v,w[i],f[u]},f[u]=tot;
		e[++tot]=edge{u,w[i],f[v]},f[v]=tot;
	}
	tarjan(1,0);
	for(int u=1;u<=n;u++)
		for(int i=f[u];i;i=e[i].next)
		{
			int v=e[i].v;
			if(col[u]!=col[v])
				g[col[u]].push_back(edge{col[v],e[i].c,0});
		}
	dfs1(rt,0);
	dfs2(rt,0);
	dfs3(rt,0);
	for(int i=1;i<=n;i++)
		printf("%lld ",dp[find(col[i])][1]);
}