codeforces 500B.New Year Permutation 解题报告

题目链接:http://codeforces.com/problemset/problem/500/B

题目意思:给出一个含有 n 个数的排列:p1, p2, ..., pn-1, pn。紧接着是一个 n * n 的矩阵A,当且仅当 Aij = 1 时,pi 与 pj 可以交换数值。现在问如何交换数值,使得最后得到的排列字典序最小。

  比赛的时候不会做,看了Tutorial 1 的解法,觉得别人做得太巧妙了,出题者也出得很好 ^_^

  可以看这个:http://codeforces.com/blog/entry/15488

    它是利用了并查集来做的。如果遇到 Aij = 1 的话,就将点 i 和 点 j 连一条边。最终会得到一些互不相交的集合。以第一组 test 来说吧~~~

  

  

  这就表示位置 1、4、7 的数是可以交换的,位置 2、5 要保持原封不动,位置 3、6 的数可以交换。由于每个集合都有一个祖先,即第一层的那个数,依次为 7、2、5、6,然后就把每个集合的数放到以该集合祖先为首的 vector 数组里面,并把每个集合内的数从小到大排序。最后遍历位置 1 ~ n,根据每个位置的数所属的集合(以哪个祖先为首),依次输出。

  只能说,解题思路真是太巧妙兼神奇!竟然可以转化成图!继续干爸爹吧 ^_^

  

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <algorithm>
 5 #include <vector>
 6 using namespace std;
 7 
 8 const int maxn = 300 + 10;
 9 int p[maxn], cnt[maxn];
10 int group[maxn];
11 char s[maxn];
12 vector<int> vi[maxn];
13 
14 int find(int x)
15 {
16     if (x == group[x])
17         return x;
18     return group[x] = find(group[x]);
19 }
20 
21 void merge(int x, int y)
22 {
23     int fx = find(x);
24     int fy = find(y);
25     if (fx != fy)
26         group[fx] = fy;
27 }
28 
29 int main()
30 {
31     #ifndef ONLINE_JUDGE
32         freopen("in.txt", "r", stdin);
33     #endif // ONLINE_JUDGE
34     int n;
35     while (scanf("%d", &n) != EOF)
36     {
37         vi[n].clear();
38         for (int i = 1; i <= n; i++) {
39             scanf("%d", &p[i]);
40             group[i] = i;
41         }
42         for (int i = 1; i <= n; i++) {
43             scanf("%s", s+1);
44             for (int j = 1; j <= n; j++) {
45                 if (s[j] == '1') {
46                     merge(i, j);
47                 }
48             }
49         }
50         for (int i = 1; i <= n; i++) {
51             vi[find(i)].push_back(p[i]);  // 千万不要写成 vi[group[i]].push_back(p[i]);
52         }
53         for (int i = 1; i <= n; i++) {
54             sort(vi[i].begin(), vi[i].end());
55         }
56         memset(cnt, 0, sizeof(cnt));
57         for (int i = 1; i <= n; i++) {
58             int g = group[i];
59             printf("%d%c", vi[g][cnt[g]++], i == n ? '
' : ' ');
60         }
61     }
62     return 0;
63 }

  注意,代码中的 51 行的部分,如果写成 vi[group[i]].push_back(p[i]) 是错的!以第二个 test 为例,group[1] = 3,find[1] = 5。find[i] 才会找到最终的祖先!!!

原文地址:https://www.cnblogs.com/windysai/p/4199621.html