标签：集训队 int ll ACM 北化 long dep now define

北化ACM集训队每周程序设计 7 问题 E: 病毒树

• 题意：给你一颗\(n\)个节点的树，有\(k\)个节点，给每个节点涂色，要求每三个相邻的节点颜色不同，求方案数。

• 题解：我们dfs这颗树，当遍历到某个节点时，要给这个点涂色，那么这个点不能和他的父亲，以及他左边的同父亲的兄弟，以及父亲的父亲同一个颜色，然后每个点能涂的颜色相乘就是方案数

• 代码：

  #include <bits/stdc++.h>
  using namespace std;
  #define PII pair<int,int>
  #define fi first
  #define se second
  #define pb push_back
  #define ll long long
  #define ull unsigned long long
  #define PLL pair<ll,ll>
  const int N=1000010;
  const int mod=1e9+7;
   
  int n;
  ll k;
  vector<int> edge[N];
  int sz[N];
  ll ans=1;
   
  void dfs(int u,int fa,int dep){
      ll now=k;
      if(dep==2) now--;
      else if(dep>=2) now-=2;
      now-=(sz[fa]-1);
      ans=ans*max(1ll*0,now)%mod;
      for(auto to:edge[u]){
          if(to==fa) continue;
          sz[u]++;
          dfs(to,u,dep+1); 
      }
  }
   
  int main(){
      scanf("%d %lld",&n,&k);
      for(int i=1;i<n;++i){
          int u,v;
          scanf("%d %d",&u,&v);
          edge[u].pb(v);
          edge[v].pb(u);
      } 
      sz[0]=1;
      dfs(1,0,1);
      printf("%lld\n",ans);
      return 0;
  }
  

标签：集训队,int,ll,ACM,北化,long,dep,now,define
来源： https://www.cnblogs.com/lr599909928/p/15549535.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。