PTA 甲级 1158 Telefraud Detection

题目链接

题目：

1.a 给 b打电话总时长<=5,则记录a打了一次短电话，若在a向b打短电话的情况下，b回了a电话，则记录a的回电次数加1；

2.若a打的短电话次数>k && a打的短电话次数>=a打的短电话中回电次数*5,则a是嫌疑人；

3.a是嫌疑人，b是嫌疑人，若a和b通话则两者可能是同一个团队。

#include<iostream>
#include<vector>
#define MAXN 1010
using namespace std;

int call[MAXN][MAXN],sc[MAXN],rec[MAXN];
vector<int> suspect;
//互相打电话，认为他们可能属于同一个团伙
int k,n,m;
int p[MAXN],vis[MAXN];

void init(){
    for(int i=0;i<MAXN;i++)p[i]=i;
}

int find(int a){
    if(a==p[a])return a;
    p[a]=find(p[a]);
    return p[a];
}

void add(int a,int b){
    int pa=find(a),pb=find(b);
    if(pa!=pb) p[pb]=pa;
}

int main(){
    init();
    cin>>k>>n>>m;
    for(int i=1;i<=m;i++){
        int x,y,t;
        cin>>x>>y>>t;
        call[x][y]+=t;//存储总通话时间
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            //一个人每天给不同的人打超过K个短电话，但==这些人==中回电的比例不超过20%
            if(call[i][j]>0&&call[i][j]<=5) {
                sc[i]++;//短电话+1
                if(call[j][i]) rec[i]++;//回电是在打电话超过k的中选出的
            }
        }
        if(sc[i]>k&&rec[i]*5<=sc[i]){
            suspect.push_back(i);
        }
    }
    for(int i=0;i<suspect.size();i++){
        for(int j=i+1;j<suspect.size();j++){
            if(call[suspect[i]][suspect[j]]&&call[suspect[j]][suspect[i]]) add(suspect[i],suspect[j]);
        }
    }
    //输出
    if(suspect.size()<=0){
        cout<<"None";return 0;
    }
    for(int i=0;i<suspect.size();i++){
        if(!vis[suspect[i]]){
            int pre=suspect[i];
            vis[pre]=1;
            cout<<pre;
            for(int j=i+1;j<suspect.size();j++){
                if(find(suspect[j]) == find(pre)){
                    cout<<" "<<suspect[j];
                    vis[suspect[j]]=1;
                }
            }
            cout<<endl;
        }
    }
    return 0;
}