树上dp:class2

类4:换根dp

题意

给定一棵n个节点的树，编号1到n，请求出以某个节点为根时，所有节点的深度之和最大。一个节点的深度定义为该节点到根节点的简单路径上边的数量。使所有节点深度之和最大的节点可能不止一个，如果有多个，则输出多个。

思路

对于无根树，可以先固定某个点为根节点，求解所有节点的深度之和怎么求解，

固定节点1为根，设g[i]表示以i节点为根节点的子树的深度之和。

设size[i]为以i为根节点的子树的节点数

则g[u]=sum(g[v]+size[v])

设f[i]为以节点i为根节点时所有节点的深度之和。

则f[v]=(f[u]-(g[v]*size[v]))+(n-size[v])

代码

#include<bits/stdc++.h>
using namespace std;
int n;
int g[1000005],f[1000005],size[1000005],vis[1000005],ans;
vector<int>ve[1000005];
void df(int u) {
    vis[u]=1;
    for(int i=0;i<ve[u].size();i++){
    	int v=ve[u][i];
        if(!vis[v]){
            df(v);
            size[u]+=size[v];
			g[u]+=(g[v]+size[v]);
			//cout<<u<<' '<<size[u]<<endl;
        }
    }
}
/*void dfs(int u) {
    vis[u]=1;
    for(int i=0;i<ve[u].size();i++){
    	int v=ve[u][i];
        if(!vis[v]){
            dfs(v);
            
            cout<<u<<' '<<g[u]<<endl;
        }
    }
}*/
void dfss(int u) {
    vis[u]=1;
    for(int i=0;i<ve[u].size();i++){
    	int v=ve[u][i];
        if(!vis[v]){
            dfss(v);
            f[v]=f[u]-2*size[v]+n;
        }
    }
}
int main(){
	cin>>n;
    int a,b;
    for(int i=1;i<=n;i++){
    	size[i]=1;
	}
    for(int i=1;i<n;i++){
        cin>>a>>b;
        ve[a].push_back(b);
        ve[b].push_back(a);
    }
    df(1);
    /*memset(vis,0,sizeof vis);
    cout<<endl;
    dfs(1);*/
    f[1]=g[1];//f[fa]-2*size[u]+n;
    memset(vis,0,sizeof vis);
    vis[1]=1;
    dfss(1);
    cout<<endl;
    int pos;
    for(int i=1;i<=n;i++){
    	cout<<f[i]<<endl;
    	if(f[i]>ans){
    		pos=i;
    		ans=f[i];
		}
	}
	cout<<pos;
	return 0;
}

类5:状压dp

位运算补充

去掉最后一位
X>>1
在最后加一个0
X<< 1
在最后加一个1
(x<< 1)+1
把最后一位变成1
x|1
最后一位取反
x^1
把右数第k位变成1
x|(1<<(k-1))
把右数第k位变成0
x&(~(1 <<(k-1)))
右数第k位取反
x^(1 << (k-1))
把最后一位变成0
(x|1)-1

题意

在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案。

国王能攻击到它上下左右，以及左上 左下右上右下八个方向上附近的各一个格子，共8个格子。

思路
f[i][j]k]表示第i行的状态为k 且已经放了j个国王的方案数
f[i][j][k]+=f[i-1][j -num[k]][p]

没有相邻国王的条件:((k&p)=0&&(x&(p<<1))==0&&(x&(p>>1))==0

代码

#include<bits/stdc++.h>
using namespace std;
int n,k;
//int pos;
long long dp[15][1000][105],num[2005];
int main(){
	cin>>n>>k;
    for(int i=0;i<(1<<n);i++){
        int tot=0,j=i;           
        while(j){
            if(j&1) tot++;
            j>>=1;
        }
        num[i]=tot;
        dp[1][i][num[i]]=1;
        //cout<<num[i]<<endl;
    }  
	for(int i=2;i<=n;i++){
		for(int j=0;j<=(1<<n)-1;j++){
                if(j&(j>>1)){
                    continue;
                }
			for(int p=0;p<=(1<<n)-1;p++){
                if(p&(p>>1)){
                    continue;
                }
				if((j&p)==0&&(j&(p<<1))==0&&(j&(p>>1))==0){
                    for(int u=num[p];u<=k;u++){
                        if(u-num[j]>=0){
                            dp[i][j][u]+=dp[i-1][p][u-num[j]];
                        }
                    }
                }
            }
		}
    }
    long long ans=0;
    for(int i=0;i<=(1<<n)-1;i++){
        ans+=dp[n][i][k];
    }
    printf("%lld\n",ans); 
	return 0;
}