题目
输入一个n行m列的数据库(内容为字符串),要求判断该数据库是否满足条件:存在列 c1 和 c2,行 r1 和 r2,使得(r1,c1)和(r2,c1)相同,(r1,c2)和(r2,c2)相同。
满足则输出三行:
NO
r1 r2
c1 c2
且前者比后者小
否则输出:YES
题解
直接枚举的话参数有四个,相当不现实。所以考虑遍历两个列,并且扫描行。扫描时用map进行储存,将此时列的元素组成二元组(pair),并且映射(map)为当前行。当遇到相同的 pair 之后输出。
还有一个技巧,就是将数据库读入的时候,转换为ID,一种字符串对应唯一ID。这个技巧核心在于对 map(str转ID) 和 vector(ID转str) 的综合运用,对于以后同样对大量字符串进行处理的时候可以用同样的技巧,因为对字符串的运算(比较)是相当慢的(当然不止只有字符串可以使用)。
代码
#include <cstdio>
#include <map>
#include <vector>
#include <set>
using namespace std;
vector<string> Strcache; //ID转string
map<string, int> IDcache; //string转ID
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;
//转换为ID,存入数据库中
if (!IDcache.count(s)) {
Strcache.push_back(s);
IDcache[s] = (int)Strcache.size() - 1;
}
base[c].push_back(IDcache[s]);
}
}
//写在函数里方便跳出
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)) mark[p] = r;
else {
printf("NO\n");
printf("%d %d\n", mark[p]+1, r+1);
printf("%d %d\n", i, j);
return;
}
}
}
printf("YES\n");
}
int main() {
int n, m;
while (scanf("%d%d", &n, &m) == 2 && n) {
vector<int> base[12]; // 10cols
IDcache.clear(); Strcache.clear();
read(base, n, m);
check(base, n, m);
}
return 0;
}
原题
UVa1592