当前位置: 代码迷 >> 综合 >> codeforces C. News Distribution
  详细解决方案

codeforces C. News Distribution

热度:29   发布时间:2023-11-19 23:31:37.0

题目链接:https://codeforces.com/contest/1167/problem/C
题目大意:有n个人,m个组。每个组中所有人都是朋友。**需要注意的是,一个人可以属于多个组。**然后,从一个人开始传递消息,他会传递给他所有的朋友,他的朋友会传递给他所有的朋友,直到朋友中没有人不知道这个消息。求这个人的消息能被多少人知道(包括自己)。求1 ~ n每个人都传递一次,每个人对应的值。
题目思路:其实很简单,就是自己bfs一次,然后看自己生成树中有多少个节点就行。首先 题目的意思就是一种广播的形式,利用bfs肯定可以得到正确结果。现在,我们需要做的就是改进。因为m组给出的信息,每组给出的都是一棵树。我们需要把这些树合并,然后得到一个n个人的关系图(图是无向的),然后bfs就行,(对于每次bfs完之后,我们可以标记已经遍历过的节点,这些节点的结果都是一样的。)bfs完之后对树中所有的节点进行赋值,值都是树的节点个数。下次遇到树中的节点就可以bfs了。对于如何生成一个完整图,我们可以将每组的边都插入到图中,而每组给出的数据应该是一个完全图,但是由于是无向图,所以我们只需要一棵树就行。因此,我们可以将第一点作为根节点,其余点都指向根。然后将每条边都加入到图中。(因为,一个点可能出现在多个组中,因此很有可能会是多重边,因此我们需要删除这些边,故利用set进行去重)
总结一下就是:
1)处理数据,将数据转化成图,用邻接链表存储,set进行去重处理。
2)执行下面的步骤:对于i,1 <= i <= n ,如果mark[i] = 0,bfs(i)。输出mark[i]。否则直接输出mark[i]。
对于bfs。普通的bfs就行,只是最后我们需要对刚刚bfs遍历过的节点进行赋值,大小就是刚刚遍历过的节点个数。

#include <bits/stdc++.h>
using namespace std;const int maxn = 500005;
int mark[maxn];
set<int> G[maxn];
void bfs(int u){vector<int> v;int cnt = 0;queue<int> q;q.push(u);mark[u] = 1;while(!q.empty()){int a = q.front();q.pop();v.push_back(a);cnt++;set<int>::iterator iter = G[a].begin();while(iter != G[a].end()){int b = *iter;if(!mark[b])q.push(b), mark[b] = 1;iter++;}}for(int i=0;i<cnt;i++){mark[v[i]] = cnt;}
}
int main(){ios::sync_with_stdio(false);int n, m;cin>>n>>m;for(int i=1;i<=n;i++)mark[i] = 0;for(int i=0;i<m;i++){int k;cin>>k;if(k == 0)continue;if(k == 1){cin>>k;continue;}int a, b; cin>>a;for(int j=1;j<k;j++){cin>>b;G[a].insert(b);G[b].insert(a);}}for(int i=1;i<=n;i++){if(!mark[i])bfs(i);cout<<mark[i]<<" ";}
}

附:这个程序处理的不太好,其实这个可以用并查集来处理。大家可以去codeforces上看并查集的处理方式。

  相关解决方案