UVA1592 数据库 Database

给出一个n行m列的数据库，是否存在两个不同行r1,r2和两个不同列c1,c2，使得这两行和这两列相同(即(r1,c1)和(r2,c1)相同，(r1,c2)和(r2,c2)相同)。

输入样例

3 3
How to compete in ACM ICPC,Peter,[e-mail]
How to win ACM ICPC,Michael,[e-mail]
Notes from ACM ICPC champion,Michael,[e-mail]
2 3
1,Peter,[e-mail]
2,Michael,[e-mail]

输出样例

NO
2 3
2 3

实现代码

#include<bits/stdc++.h>
using namespace std;
vector<string> strCont;//用于存储字符串，为其编号
map<string, int> strAndID;//字符串和编号一一对应
int n, m;
//读取字符串，进行初始化
void read(vector<int> base[], int n, int m) {
  for (int i = 1; i <= 10; i++) {
    base[i].clear();//置空
  }
  getchar();//获取上一行的‘\n’
  for(int r = 1; r <= n; r++)
    for (int c = 1; c <= m; c++) {
      string s = "";
      char ch;
      while ((ch = getchar()) != ',' && ch != '\n')
        s += ch;
      if (!strAndID.count(s)) {
        strCont.push_back(s);
        strAndID[s] = strCont.size() - 1;
      }
      base[c].push_back(strAndID[s]);//数据库最终保存的是列c的各行字符串对应的ID.  由于题中行数较多,所以使用用列数然后在上面添加元素.也可以使用一个二维数组,不过内存开销就大了.
    }
}
//寻找符合条件的字符串.
void check(vector<int> base[], int n, int m) {
  //使用i 和j遍历列,寻找符合题意的
  for (int i = 1; i <= m; i++) {
    for (int j = i + 1; j <= m; j++) {
      map<pair<int, int>, int> mark;
      //扫描行,把数据和行号存起来
      for (int r = 0; r < base[i].size(); r++) {
        pair<int, int> p(base[i][r], base[j][r]);//生成二元组
        if (!mark.count(p)) {//判断之前是否已经出现过这样的二元组 p
          mark[p] = r;//没出现过就存着,然后标号
        }
        else {
          cout << "NO" << endl;
          cout << mark[p] + 1 << " " << r + 1 << endl;//输出行号
          cout << i << " " << j << endl;//输出列号
          return;//找到就结束
        }
      }
    }
  }
  cout << "YES" << endl;//没找到就输出YES.
}

int main()
{
  while (cin >> n >> m && n) {
    vector<int> base[12];//用于存数据库  10cols
    strAndID.clear();//初始化
    strCont.clear();
    read(base, n, m);
    check(base, n, m);
  }
  return 0;
}