Codeforces Round #541 (Div. 2)D. Gourmet choice详解

Codeforces Round #541 (Div. 2)D. Gourmet choice详解

题目链接
题目大意：
美食家第一天吃了n盘菜，第二天吃了m盘菜，美食家将两天的菜的美味程度比较写成n*m的方阵.
a[i][j]表示第一天第i盘菜和第二天第j盆菜的比较。请你给每盆菜打分，使每盆菜的分数尽量小，，且最小为1；
如果表格中关系矛盾输出“No”，否则输出“Yes”，并在第一第二行分别输出第一第二天个盆菜的平分；
思维：
由比较关系输出各盆菜的最小分数，很容易 想到拓扑排序；小蒟蒻溜了溜了
实则拓扑排序听着高大尚，其实思想很简单。
记录每个数大于的数的个数，然后按每个数大于的数的个数入队；
不是很理解的同学向大家推荐大佬的博客“戳这”相信大家很容易理解。

此题还有一个妨碍拓扑排序的障碍即有‘=’，这个时候我们就可以用并查集进行缩点操作。关于并查集的用法此处就不再赘述；并查集详解“戳这”

其余的具体操作将在代码中详细讲解

#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
typedef pair<int, int> pii;
char s[1100][1100];
int f[2100], level[2100], res[2100], result[2100];
vector<int>bigger[2100];
queue <pii> q;
int find(int x)
{
 while (x != f[x])
  x = f[x] = f[f[x]];
 return x;
}
int main()
{
 int n, m;
 cin >> n >> m;
 for (int i = 1; i <= n + m; i++)
  f[i] = i;
 for (int i = 1; i <= n; i++)
  for (int j = 1; j <= m; j++)
   cin >> s[i][j];
   //--------------利用并查集合并相同点
 for (int i = 1; i <= n; i++)
  for (int j = 1; j <= m; j++)
   if (s[i][j] == '=')
    f[find(i)] = find(j + n);
    //------------------统计比每个数小的点的个数和记录比每个数大的点
 for (int i = 1; i <= n; i++)
 {
  for (int j = 1; j <= m; j++)
  {
   if (s[i][j] == '=')
    continue;
   int ei = find(i), ej = find(j + n);
   if (s[i][j] == '<')
   {
    bigger[ei].push_back(ej);
    level[ej]++;
   }
   else
   {
    bigger[ej].push_back(ei);
    level[ei]++;
   }
  }
 }
 //---------------开始拓扑排序
 for (int i = 1; i <= n + m; i++)
  if (level[i]==0)//此处不需要剔除相等的点，在接下来的操作中会解决
   q.push(pii(i, 1));
 while (!q.empty())
 {
  pii p = q.front();
  q.pop();
  res[p.first] = p.second;
  for (int i=0; i<bigger[p.first].size(); i++)
  {
   if (--level[bigger[p.first][i]] == 0)
    q.push(pii(bigger[p.first][i], p.second + 1));
  }
 }
 for (int i = 1; i <= n + m; i++)
  result[i] = res[find(i)];//将相等的点相互匹配
  //-----------------判断生成比分是否有所给表格矛盾
 for (int i = 1; i <= n; i++)
 {
  for (int j = 1; j <= m; j++)
  {
   if (s[i][j] == '='&&result[i] != result[j + n] ||
    s[i][j] == '>'&&result[i] <= result[j + n] ||
    s[i][j] == '<'&&result[i] >= result[j + n])
   {
    cout << "No\n";
    return 0;
   }
  }
 }
 //-------------按要求输出，收工
 cout << "Yes\n";
 for (int i = 1; i <= n; i++)
  cout << result[i] << ' ';
 cout << '\n';
 for (int j = n + 1; j <= n + m; j++)
  cout << result[j] << ' ';
 cout << '\n';
}