hdu5977 Garden of Eden

题目:点击打开链接

给你一棵n个节点的树,一共有k种颜色,每个节点有一种颜色,求a到b的路径中经过了所有颜色的路径的对数,(a,b)和(b,a)都要算

思路:树分治+状态压缩+枚举子集

代码:

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define PI acos(-1.0)
#define LL long long
#define ULL unsigned long long
#define INF 0x3f3f3f3f
#define mm(a,b) memset(a,b,sizeof(a))
#define PP puts("*********************");
template T f_abs(T a){ return a > 0 ? a : -a; }
template T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template T lcm(T a,T b){return a/gcd(a,b)*b;}
// 0x3f3f3f3f3f3f3f3f
// 0x3f3f3f3f

const int maxn=5e4+50;
struct Node{
    int v,next;
}edge[2*maxn];
int head[maxn],tol;//节点从1开始计数
int son[maxn],f[maxn],vis[maxn];
//son表示以i为根的子树大小
//f数组表示以u为根的最大子树的大小
//vis表示该节点是不是已经作为重心了
int d[maxn],arr[maxn];
int cntv,root,K,ALL;
LL ans,num[1050],sum[1050];
vector g[1050];
void init(){
    mm(head,-1);tol=0;
    mm(vis,0);
    f[0]=INF;//一定要初始化成一个很大的值
}
void addedge(int u,int v){
    edge[tol].v=v;
    edge[tol].next=head[u];
    head[u]=tol++;
}
void getroot(int u,int fa){//寻找重心
    son[u]=1,f[u]=0;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v==fa||vis[v]) continue;
        getroot(v,u);
        son[u]+=son[v];
        f[u]=max(f[u],son[v]);
    }
    f[u]=max(f[u],cntv-son[u]);//sum表示当前树的大小
    if(f[u]


你可能感兴趣的:(点分治)