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

题意：中文题面自行理解

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

代码：

#include
using namespace std;
#define LL long long
const int maxn=1e5+10;
const LL mod=1e9+7;
int n,k;
map mp;
vector vec[maxn],tmp;///tmp:所有只有红色边连成的连通块的点个数
int book[maxn];///标记点是否只有黑边与之相连
int fun(int x)
{
    mp[x]=1;
    queue que;
    int ans=1;
    que.push(x);
    while(que.size()){
        int p=que.front();
        que.pop();
        for(int i=0;i