快手2020校园招聘秋招笔试--工程A试卷题4

题意：中文题面自行理解

思路：处理出每块由红色边连成的连通块的点个数，ans=总方案数-所有只有红边相连的块的总方案数+只有黑边与之相连的点个数

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int maxn=1e5+10;
const LL mod=1e9+7;
int n,k;
map<int,int> mp;
vector<int> vec[maxn],tmp;///tmp:所有只有红色边连成的连通块的点个数
int book[maxn];///标记点是否只有黑边与之相连
int fun(int x)
{mp[x]=1;queue<int> que;int ans=1;que.push(x);while(que.size()){int p=que.front();que.pop();for(int i=0;i<vec[p].size();i++){if(mp[vec[p][i]]==0){que.push(vec[p][i]);mp[vec[p][i]]=1;ans++;}}}return ans;
}
LL cal(int x)
{x=(LL)x;LL ans=1ll;for(int i=1;i<=k;i++){ans=(ans*x)%mod;}return ans;
}
int main()
{scanf("%d%d",&n,&k);int u,v,w;for(int i=1;i<n;i++){scanf("%d%d%d",&u,&v,&w);if(w==0){vec[u].push_back(v);vec[v].push_back(u);book[v]=1;book[u]=1;}}for(int i=1;i<=n;i++){for(int j=0;j<vec[i].size();j++){if(mp[vec[i][j]]) continue;tmp.push_back(fun(vec[i][j]));}}LL ans=cal(n);for(int i=1;i<=n;i++){if(book[i]==0){ans=(ans-1+mod)%mod;}}for(int i=0;i<tmp.size();i++){ans=(ans-cal(tmp[i])+mod)%mod;}printf("%lld\n",ans);return 0;
}

快手2020校园招聘秋招笔试--工程A试卷题4相关推荐

快手2020校园招聘秋招笔试--工程B试卷
攻击者使用无效IP地址,利用TCP连接的三次握手过程,连续发送会话请求,使受害主机处于开放会话的请求之中,直至连接超时,最终因耗尽资源而停止响应.这种攻击被称为 A. DNS欺骗攻击 B. DDoS攻 ...
快手2020校园招聘秋招笔试--工程C试卷 (编程题题解全)
1.病毒检测滑动窗口题例子: 2 0101010(第一位下标为1) 1)找到第一个最短的满足k条件的子串,即s[2~4],l=2,r=4 2)找出这个子串左右连续的0的个数,分别为a,b,则对于这 ...
快手2020校园招聘秋招笔试--工程A试卷 (编程题题解全)
"好序列"的个数反着求,先求出所有序列个数,再减去不符合的个数. 黑边把图分成一个,一个的连通分量. 所以我们只要用dfs求出每个不包含黑边的连通分量的包含点个数sz,sz^k就 ...
快手2020校园招聘秋招笔试--工程A试卷
1.(计算机网络原理)若用斜杠记法标识子网掩码,则255.255.240.0对应于 B A. /19 B. /20 C. /21 D. /22 解析 255.255.240.0转换为二进制为11111 ...
快手2020校园招聘秋招笔试--工程B试卷订正
1.三种攻击 DNS欺骗攻击:冒充域名把原来查询的IP改为其他IP ddos攻击:大量合法的服务器向某一目标不断发送请求,导致其他用户无法使用 SYN Flooding攻击:攻击者使用无效IP地址 ...
快手2020校园招聘秋招笔试--工程C试卷
1. 关于java的异常处理机制,以下说法正确的是: A. 当某个线程抛出OutOfMemoryError时,其他线程有可能不受影响 B. 当大量抛出RuntimeException时,不会影响系统的 ...
快手2020校园招聘秋招笔试--工程C试卷（21题）滑动窗口解决字串
病毒检测时间限制:C/C++ 1秒,其他语言2秒空间限制:C/C++ 32M,其他语言64M 小明最近在做病毒自动检测,他发现,在某些library 的代码段的二进制表示中,如果包含子串并且恰好有 ...
渣渣做后端真题-- 快手2020校园招聘秋招笔试--工程A试卷选择题
真的是一枚渣渣,题在牛客网找的,有错误求大神指导,求一起做题的小伙伴,没写的题是我会的,有问题可以给我留言呀- A:记忆小Tips:4** 没有402(阿玛尼红管402 ,一只鲜艳的苹果红),5** ...
快手2020校园招聘秋招笔试--算法B试卷
四个编程题总体算比较简单,基本都在LeetCode出现过.但是我的速度太慢了,还要加油啊ヾ(◍°∇°◍)ﾉﾞ牛客快手2020校园招聘秋招笔试–算法B试卷合法数独给定一个数独板的输入,确认当前的填 ...

快手2020校园招聘秋招笔试--工程A试卷题4

快手2020校园招聘秋招笔试--工程A试卷题4相关推荐

最新文章

热门文章

快手2020校园招聘秋招笔试--工程A试卷 题4

快手2020校园招聘秋招笔试--工程A试卷 题4相关推荐

最新文章

热门文章

快手2020校园招聘秋招笔试--工程A试卷题4

快手2020校园招聘秋招笔试--工程A试卷题4相关推荐