北化ACM集训队每周程序设计 7 问题 E: 病毒树
2021/11/13 20:40:19
本文主要是介绍北化ACM集训队每周程序设计 7 问题 E: 病毒树,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
北化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; }
这篇关于北化ACM集训队每周程序设计 7 问题 E: 病毒树的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-10-07fastcgi 是什么-icode9专业技术文章分享
- 2024-10-07fastcgi 的详细使用教程介绍-icode9专业技术文章分享
- 2024-10-07git如何更新单个文件到本地-icode9专业技术文章分享
- 2024-10-07如何使用ASM(Abstract Syntax Tree Manipulation)技术来修改第三方AAR依赖中的函数-icode9专业技术文章分享
- 2024-10-07Activity 跳转时间耗时很长怎么优化解决-icode9专业技术文章分享
- 2024-10-07Androud Toast 有哪些常用的第三方组件-icode9专业技术文章分享
- 2024-10-07在viewmodel中怎么使用 mmkv?-icode9专业技术文章分享
- 2024-10-07MMKV.defaultMMKV() 是单例模式吗?-icode9专业技术文章分享
- 2024-10-04el-table 开启定时器下,表格的选中状态会消失是什么原因-icode9专业技术文章分享
- 2024-10-03如何安装和初始化飞牛私有云 fnOS?-icode9专业技术文章分享