仅验证了样例,而且感觉自己时间复杂度写的有点高
这道题说的是找telefraud,并且将他们分别放进所属的gang中,其中telefraud是有超过k的不同的人通电话时长少于5min,并且这些通话时长少于5min的人中少于20%的人给他回了电话。而gang是已经是telefraud的人中有相互通过话的。感觉这道题因为题意的关系还是挺绕的。
首先用unordered_set记录那些通话时间小于5min的人,这个容器可以帮我们自动去重,可以直接得到通话时间小于5min的人的列表,同时用unordered_map来记录一下谁互通过电话。之后看看对于每个人的回话有多少人,如果达到了我们的要求,那么这个人就是实打实的telefraud。在这里,为了之后的输出格式,要把telefraud们按照编号排一下序。
之后要把这些人归到各个gang里,这里用了并查集,因为map会按照key值排序,因此只需要让已经排好序的每个gang里的telefraud的father是最小编号的telefraud,因此将Union和找gang的函数的顺序按照下面代码这样写了(也就是father[faB]=faA;同时在填写时写Union(i,j))。之后用迭代器对map里的元素进行输出就可以了
#include <iostream>
#include <vector>
#include <algorithm>
#include <unordered_set>
#include <unordered_map>
#include <map>
using namespace std;
const int maxv=1010;
unordered_set<int> se[maxv];
unordered_map<int,int> mp;
vector<int> telefraud;
int father[maxv];
void init(){
for(int i=0;i<maxv;i++){
father[i]=i;
}
}
int findfather(int x){
int a=x;
while(x!=father[x]){
x=father[x];
}
while(a!=father[a]){
int z=a;
a=father[a];
father[z]=x;
}
return x;
}
void Union(int a,int b){
int faA=findfather(a);
int faB=findfather(b);
if(faA!=faB){
father[faB]=faA;
}
}
int main(){
int k,n,m;
cin>>k>>n>>m;
init();
for(int i=0;i<m;i++){
int v1,v2,len;
scanf("%d %d %d",&v1,&v2,&len);
mp[v1*1000+v2]=1;
if(len<=5){
se[v1].insert(v2);
}
}
for(int i=1;i<=n;i++){
if(se[i].size()>k){
int callback=0;
for(auto it=se[i].begin();it!=se[i].end();it++){
if(mp[*it*1000+i]==1){
callback++;
}
}
int nomorethan=(int)se[i].size()*0.2;
if(callback<=nomorethan){
telefraud.push_back(i);
}
}
}
if(telefraud.size()==0){
printf("None\n");
}else{
sort(telefraud.begin(),telefraud.end());
for(int i=0;i<telefraud.size()-1;i++){
for(int j=i+1;j<telefraud.size();j++){
if(mp[telefraud[i]*1000+telefraud[j]]==1&&mp[telefraud[j]*1000+telefraud[i]]==1){
Union(telefraud[i],telefraud[j]);
}
}
}
map<int,vector<int>> gang;
for(int i=0;i<telefraud.size();i++){
int father=findfather(telefraud[i]);
gang[father].push_back(telefraud[i]);
}
for(auto it=gang.begin();it!=gang.end();it++){
for(int i=0;i<it->second.size();i++){
printf("%d",it->second[i]);
if(i!=it->second.size()-1){
printf(" ");
}else{
printf("\n");
}
}
}
}
return 0;
}