AtCoder Beginner Contest 341-F

AtCoder Beginner Contest 341-F

F - Breakdown

Problem

给你一个由 $N$ 个顶点和 $M$ 条边组成的简单无向图。每个顶点拥有权重$W_i$，并且被放置了$A_i$个棋子。

只要图形上还有棋子，就重复下面的操作：

• 首先，从图形中选择一个（有棋子的）顶点$x$并移除一个棋子。
• 从$x$相邻点中选择出一些点组成集合$S$（可以不选），要保证这个集合内的所有点的权重之和小于顶点$x$，即${\sum }_{y\in S}{W}_y<W_x$，并在$S$中的每个顶点上放置一个棋子。

请求出最多最多能进行多少次这样的操作。

可以证明，无论如何操作，在有限次迭代后，图形上将没有棋子。

Constraints

• $2\le N\le 5000$
• 1 ≤ M ≤ min ⁡ { N ( N − 1 ) / 2 , 5000 } 1 \leq M \leq \min \lbrace N(N-1)/2, 5000 \rbrace 1≤M≤min{N(N−1)/2,5000}
• $1\le u_i,v_i\le N$
• $u_i\ne v_i$
• i ≠ j    ⟹    { u i , v i } ≠ { u j , v j } i \neq j \implies \lbrace u_i, v_i \rbrace \neq \lbrace u_j, v_j \rbrace i=j⟹{ui​,vi​}={uj​,vj​}
• $1\le W_i\le 5000$
• $0\le A_i\le 10^9$

Solution

首先再此解释一下题目中的操作：

假设现在图是这样的：

（为了方便，图中的数字既表示顶点，同时也表示该点的权重$W_i$）

其中顶点5上有一颗棋子，并且现在选择$x=5$，开始操作。

首先取下5上的棋子，接下来选择5的相邻点的一个集合（比如1,3），且保证集合内点的权重之和小于$W_x$。因此我们可以选1,3，可以选1，可以选3，也可以不选，但是不能选6。

此时可以发现，我们始终只能选择比点$x$要小的点，也就是说棋子的扩散方向永远是单向的（3永远不可能给5棋子，5也永远不能给6棋子）

所以题目中说的简单无向图其实是个幌子，这其实是一个DAG

那么我们先将原图化为DAG，再按照权重升序来看各个顶点（小权重顶点不会对大权重顶点有干扰，棋子之间也不会有相互作用），进行DAG上的dp。

具体的，设X[i]表示若顶点i有一枚棋子，可以操作的次数

当轮到点x的时候，权重比其小的出点都已经计算好了X[i]，现在需要抉择出如何选择他的出点集合，使得在权重之和不超过$W_x$的情况下，尽力包含更多的X[i]——相当于做个背包问题了。

Code

#define N 5020


int n,m;
vector<pair<int,int>>edge;
int W[N],A[N];
pair<int,int>ww[N];
vector<int>e[N];
int order[N];


bool cmp(int x,int y)
{
	return W[x]<W[y];
}

LL dp[N];
LL X[N];

int main()
{
	cin>>n>>m;
	for(int i=0;i<m;i++)
	{
		int x,y;
		cin>>x>>y;
		edge.push_back(make_pair(x,y));
	}
	for(int i=1;i<=n;i++)
	{
		cin>>W[i];
	}
	for(int i=1;i<=n;i++)
	{
		cin>>A[i];
	}
	for(int i=0;i<m;i++)
	{
		if(W[edge[i].first]>W[edge[i].second]) e[edge[i].first].push_back(edge[i].second);
		if(W[edge[i].first]<W[edge[i].second]) e[edge[i].second].push_back(edge[i].first);
	}
	for(int i=1;i<=n;i++) order[i]=i;
	sort(order+1,order+n+1,cmp);
	for(int i=1;i<=n;i++) X[i]=1;
	for(int i=1,x;i<=n;i++)
	{
		x=order[i];
		memset(dp,0,sizeof(dp));
		dp[0]=1;
		for(unsigned int j=0;j<e[x].size();j++)
		{
			int y=e[x][j];
			for(int k=W[x]-1;k-W[y]>=0;k--)
			{
				dp[k]=max(dp[k],dp[k-W[y]]+X[y]);
			}
		}
		for(int k=0;k<=5000;k++) X[x]=max(X[x],dp[k]);
	}
	LL ans=0;
	for(int i=1;i<=n;i++) ans+=X[i]*A[i];
	cout<<ans;
	
	
	return 0;
}

Attention

记得开long long

注意背包dp[i]与X[i]的初值

Reference

A-F in 4 minutes