codeforces 379G. New Year Cactus

题目大意:给一颗2500点的仙人掌,每个点可以染黑色,白色,不染色。同一条边两端点不能为黑白,求对于所有黑点个数的白点个数的最大值。

我直接想到了把他的dfs树建出来,直接dp节点x的儿子有y个黑点,当前染a,子树的返祖边染b的最大值。但是我以为他是O(n^3)的。但实际上他好像是O(n^2),然后就没有然后了。

//具体怎么推我真的不会,大概有这样的式子吧:T(n)=T(a)+T(n-a)+O(n*min(a,n-a)),反正平衡的二叉树是对的。。。

dp写的超长。。。

#include
#define N 2505
using namespace std;
int n,m,x,y,dp[N][N][3][3],size[N],top[N],vis[N],fl[N],fa[N],dep[N];
int fst[N],to[N*4],nxt[N*4],l;

//dp[x] [y][a][b]
void link(int x,int y)
{
	to[++l]=y;nxt[l]=fst[x];fst[x]=l;
	to[++l]=x;nxt[l]=fst[y];fst[y]=l;
}
void upd(int &x,int y)
{
	if (y>x) x=y;
}
//0 1 2
void merge(int a,int b)
{
	int n,m;
	n=size[a];m=size[b];
	static int tmp[N][3][3];
	for (int i=0;i<=n+m;i++)
	for (int j=0;j<3;j++)
	for (int k=0;k<3;k++)
		tmp[i][j][k]=-1000000;
	for (int i=0;i<=n;i++)
	for (int j=0;j<=m;j++)
	{
		if (top[a]==top[b]&&top[b])
		{
			for (int x=0;x<3;x++)
			{
				upd(tmp[i+j][0][x],dp[a][i][0][0]+dp[b][j][0][x]);
				upd(tmp[i+j][0][x],dp[a][i][0][0]+dp[b][j][1][x]);
				upd(tmp[i+j][0][x],dp[a][i][0][0]+dp[b][j][2][x]);
			}
			for (int x=1;x<3;x++)
			{
				for (int y=0;y<3;y++)
				{
					upd(tmp[i+j][x][y],dp[a][i][x][0]+dp[b][j][0][y]);
					upd(tmp[i+j][x][y],dp[a][i][x][0]+dp[b][j][x][y]);
				}
			}
		}
		if (a==top[b])
		{
			for (int x=0;x<3;x++)
			{
				upd(tmp[i+j][0][0],dp[a][i][0][0]+dp[b][j][0][x]);
				upd(tmp[i+j][0][0],dp[a][i][0][0]+dp[b][j][1][x]);
				upd(tmp[i+j][0][0],dp[a][i][0][0]+dp[b][j][2][x]);
			}
			for (int x=1;x<3;x++)
			{
				upd(tmp[i+j][x][0],dp[a][i][x][0]+dp[b][j][0][0]);
				upd(tmp[i+j][x][0],dp[a][i][x][0]+dp[b][j][x][0]);
				upd(tmp[i+j][x][0],dp[a][i][x][0]+dp[b][j][0][x]);
				upd(tmp[i+j][x][0],dp[a][i][x][0]+dp[b][j][x][x]);
			}
		}
		if (top[a]==0&&top[b]==0)
		{
			upd(tmp[i+j][0][0],dp[a][i][0][0]+dp[b][j][0][0]);
			upd(tmp[i+j][0][0],dp[a][i][0][0]+dp[b][j][1][0]);
			upd(tmp[i+j][0][0],dp[a][i][0][0]+dp[b][j][2][0]);
			for (int x=1;x<3;x++)
			{
				for (int y=0;y<3;y++)
				{
					upd(tmp[i+j][x][y],dp[a][i][x][0]+dp[b][j][0][y]);
					upd(tmp[i+j][x][y],dp[a][i][x][0]+dp[b][j][x][y]);
				}
			}
		}
		if (top[a]&&top[a]!=top[b])
		{
			for (int x=0;x<3;x++)
			{
				upd(tmp[i+j][0][x],dp[a][i][0][x]+dp[b][j][0][0]);
				upd(tmp[i+j][0][x],dp[a][i][0][x]+dp[b][j][1][0]);
				upd(tmp[i+j][0][x],dp[a][i][0][x]+dp[b][j][2][0]);
			}
			for (int x=1;x<3;x++)
			{
				for (int y=0;y<3;y++)
				{
					upd(tmp[i+j][x][y],dp[a][i][x][y]+dp[b][j][0][0]);
					upd(tmp[i+j][x][y],dp[a][i][x][y]+dp[b][j][x][0]);
				}
			}
		}
	}
	for (int i=0;i<=n+m;i++)
	for (int j=0;j<3;j++)
	for (int k=0;k<3;k++)
		dp[a][i][j][k]=tmp[i][j][k];
}
void dfs(int x)
{
	vis[x]=1;size[x]=1;dep[x]=dep[fa[x]]+1;
	dp[x][0][0][0]=0;
	dp[x][0][1][0]=1;
	dp[x][0][2][0]=-1000000;
	
	dp[x][1][0][0]=-1000000;
	dp[x][1][1][0]=-1000000;
	dp[x][1][2][0]=0;
	int tmp=0;
	for (int i=fst[x];i;i=nxt[i])
	if (!vis[to[i]])
	{
		fa[to[i]]=x;
		dfs(to[i]);
		if (top[to[i]]&&top[to[i]]!=x)
			tmp=to[i];
		else
		{
			merge(x,to[i]);
			size[x]+=size[to[i]];
		}
	}
	else
	{
		if (dep[to[i]]


你可能感兴趣的:(codeforces 379G. New Year Cactus)