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;
}