PATA1139题解

最新推荐文章于 2022-03-17 20:06:18 发布

原创最新推荐文章于 2022-03-17 20:06:18 发布 · 496 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#PAT

PAT 专栏收录该内容

131 篇文章

订阅专栏

本文介绍了一个解决PATA1139编程问题的方法，该问题关注于通过读取特定格式的数据来识别和匹配社交网络中的朋友关系。文章详细解释了如何处理特殊输入格式（如-0000），并确保输出符合题目要求（例如四位数的朋友编号）。通过使用C++实现的算法，有效地解决了性别匹配和非直接朋友间的连接问题。

三个坑点：

1、测试数据中有-0000

2、输出的朋友对为四位数

3、当a，b性别相同时，要判断a的朋友c不为b， b的朋友d不为a

//
//  main.cpp
//  PATA1139
//
//  Created by Phoenix on 2018/2/26.
//  Copyright © 2018年 Phoenix. All rights reserved.
//

#include <iostream>
#include <cstdio>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 10010;
int gender[maxn];   //1为女性，2为男性
int G[maxn][maxn];      //标记两人是否为朋友，是为1，否为0
vector<int> v[maxn];

struct node{
    int first, second;      //记录输出的两个朋友
};

bool cmp(node a, node b) {      //将输出的朋友对排序
    if(a.first != b.first) return a.first < b.first;
    else return a.second < b.second;
}

int main(int argc, const char * argv[]) {
    int n, m, k;
    scanf("%d %d", &n, &m);
    for(int i = 0; i < m; i++) {
        char s1[6], s2[6];  //通过字符串输入排除-0000的情况
        int a = 0, b = 0, flag1, flag2;
        scanf("%s %s", s1, s2);
        if(s1[0] == '-') flag1 = 1;     //如果有负号则为女性1
        else flag1 = 2;
        if(s2[0] == '-') flag2 = 1;
        else flag2 = 2;
        for(int i = 0; i < strlen(s1); i++) {   //a， b 都转化为自然数
            if(s1[i] != '-') {
                a = a * 10 + s1[i] - '0';
            }
        }
        for(int i = 0; i < strlen(s2); i++) {
            if(s2[i] != '-') {
                b = b * 10 + s2[i] - '0';
            }
        }
        gender[a] = flag1;      //标记a， b 的性别
        gender[b] = flag2;
        G[a][b] = G[b][a] = 1;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    scanf("%d", &k);
    for(int i = 0; i < k; i++) {
        int a, b, gen_1, gen_2;
        scanf("%d %d", &a, &b);
        if(a < 0) {     //将a，b转化为自然数
            a *= -1;
        }
        if(b < 0) {
            b *= -1;
        }
        gen_1 = gender[a];
        gen_2 = gender[b];
        vector<node> v1;    //保存满足条件的朋友对
        for(int j = 0; j < v[a].size(); j++) {
            if(gender[v[a][j]] == gen_1 && v[a][j] != b) {      //寻找a的同性朋友且这个朋友不为b
                for(int s = 0; s < v[b].size(); s++) {
                    if(gender[v[b][s]] == gen_2 && G[v[b][s]][v[a][j]] == 1 && v[b][s] != a) {   //寻找b的同性朋友v[b][s]且该朋友与a的朋友v[a][j]互为朋友，且b的同性朋友不为a
                        node no;
                        no.first = v[a][j];
                        no.second = v[b][s];
                        v1.push_back(no);       //将满足条件的朋友对加入v1中
                    }
                }
            }
        }
        sort(v1.begin(), v1.end(), cmp);    //将待输出朋友对排序
        if(v1.size() == 0) printf("0\n");
        else {
            printf("%d\n", v1.size());
            for(int j = 0; j < v1.size(); j++) {
                printf("%04d %04d\n", v1[j].first, v1[j].second); //注意输出为四位数
            }
        }
    }
    return 0;
}