SDUT 3811 离散题目17

Problem Description

给出集合X和X上的关系R，求关系R在X上的对称闭包s(R)。

例如：

X={1,2,3,4,5} , R={<1,1>,<2,1>,<3,3>,<2,3>,<3,2>,<4,5>}

s(R)= {<1,1>,<1,2>,<2,1>,<2,3>,<3,2>,<3,3>,<4,5>,<5,4>}

Input

多组输入，每组输入第一行为集合X的元素；第二行为一个整数n ( n > 0 )，代表X上的关系R中序偶的个数；接下来n行用来描述X上的关系R，每行两个数字，表示关系R中的一个序偶。细节参考示例输入。

非空集合X的元素个数不大于500，每个元素的绝对值不大于2^32 - 1。

Output

每组输入对应一行输出，为X上关系R的对称闭包s(R)，s(R)中的序偶根据序偶中的第一个值升序排列，如果第一个值相同则根据第二个值升序排列；具体输出格式见样例（注意：样例每个逗号后有一个空格）。

Example Input

1 2 3 4 5
6
1 1
2 1
3 3
2 3
3 2
4 5

Example Output

[(1, 1), (1, 2), (2, 1), (2, 3), (3, 2), (3, 2), (3, 3), (4, 5), (5, 4)]

代码: 求对称闭包，然而这个输出样例是有问题的
正确的应该是：[(1, 1), (1, 2), (2, 1), (2, 3), (2, 3), (3, 2), (3, 2), (3, 3), (4, 5), (5, 4)]

#include<bits/stdc++.h>using namespace std;struct node{    int x, y;    bool operator < (const node &b) const {        if(x == b.x) return y < b.y;        else return x < b.x;    }};vector<node> a;int main(){    string s;    int n, i, u, v;    while(getline(cin, s))    {        scanf("%d", &n);        for(i = 0; i < n; i++)        {            scanf("%d %d", &u, &v);            a.push_back((node){u, v});//存起来        }        //getchar();        sort(a.begin(), a.end());//排序        n = a.size();        for(i = 0; i < n; i++)        {            if(a[i].x != a[i].y)//如果不相等，交换一下存入            {                a.push_back((node){a[i].y, a[i].x});            }        }        sort(a.begin(), a.end());//排序，然后输出        printf("[(%d, %d)", a[0].x, a[0].y);        for(i = 1; i < a.size(); i++)        {            printf(", (%d, %d)", a[i].x, a[i].y);        }        printf("]\n");        a.clear();        getline(cin, s);//坑点，数据后面有多余的空格    }    return 0;}