洛谷2016 战略游戏(树形DP)

传送门

【题目分析】

之前觉得这种入门DP很难来着qwq

由于n很小,加上可以记忆化,所以直接暴力搜索,枚举父亲是否放了,如果没放那么当前点只能选择放,否则两种都可以,这样dfs下去统计答案即可。

【代码~】

#include
using namespace std;
const int MAXN=2e3+10;
const int MAXM=4e3+10;

int n,cnt;
int a[MAXN],dp[MAXN][2];
int head[MAXN];
int nxt[MAXM],to[MAXM];

int Read(){
	int i=0,f=1;
	char c;
	for(c=getchar();(c>'9'||c<'0')&&c!='-';c=getchar());
	if(c=='-')
	  f=-1,c=getchar();
	for(;c>='0'&&c<='9';c=getchar())
	  i=(i<<3)+(i<<1)+c-'0';
	return i*f;
}

void add(int x,int y){
	nxt[cnt]=head[x];
	head[x]=cnt;
	to[cnt]=y;
	cnt++;
}

int dfs(int u,int fa,int used){
	if(dp[u][used]!=0x3f3f3f3f)
	  return dp[u][used];
	int ret=0;
	for(int i=head[u];i!=-1;i=nxt[i]){
		int v=to[i];
		if(v==fa)
		  continue;
		if(!used){
			ret+=dfs(v,u,1)+1;
		}
		else{
			ret+=min(dfs(v,u,0),dfs(v,u,1)+1);
		}
	}
	dp[u][used]=ret;
	return ret;
}

int main(){
	memset(head,-1,sizeof(head));
	memset(dp,0x3f,sizeof(dp));
	n=Read();
	for(int i=1;i<=n;++i){
		int x=Read(),k=Read();
		for(int j=1;j<=k;++j){
			int y=Read();
			add(x,y),add(y,x);
		}
	}
	dfs(0,-1,0);
	dfs(0,-1,1);
	cout<

 

你可能感兴趣的:(————DP————)