文章目录
链接Teach Me
题目大意:N个员工,编号1到N,2 ≤ N ≤ 5 × 10^4. S项技能,编号1到S, 1 ≤ S ≤ 1000。员工i有技能集skills[i],1 ≤ 个数 ≤ 5。如果员工i有一项技能是员工j所没有的,那i可以教j。求有多少个有序对(i, j)使得i可以教j。
Approach 1
固定i,看员工i能教多少人,然后都加起来。
伪码
res = 0
for i in [1, N]:
hashset s
for x in skills[i]:
for j in [1, N]:
if j has skill x:
s.insert(j)
res += s.size()
return res
时间复杂度O(N^2) TLE
Approach 2
从反面来想,如果i不能教j会怎么样。i不能教j,当且仅当skills[i]是skills[j]的子集。通俗地来讲,如果i会的,j全都会,那i拿什么来教j。
固定j,枚举skills[j]的子集,小于2^5个。如果有x个员工的技能与某子集s相等,那这x个员工不能教j。为了加快集合的比较,使用一个hashmap来记录每种集合出现的次数,可以将集合转换成字符串,也可以当成一个10001进制,位数不大于5的数。时间复杂度O(N)
#include <iostream>
#include <vector>
#include <unordered_map>
#include <string>
#include <algorithm>
using namespace std;
const int MAXN = 5e4;
int skls[MAXN][6]; // skls[i]表示人i拥有的技能,skls[i][0]是数量
class Solution {
public:
void solve() {
int N, S, C;
unordered_map<string, int> cnt;
cin >> N >> S;
for(int j = 0; j < N; j++) {
cin >> C;
skls[j][0] = C;
string key;
// 可以改成10001进制的C位数
for(int k = 1; k <= C; k++) {
cin >> skls[j][k];
}
// 技能要先排序,才能保证相等的集合转化成相等的键
sort(skls[j]+1, skls[j]+C+1);
for(int k = 1; k <= C; k++) {
key += to_string(skls[j][k]) + " ";
}
cnt[key]++;
}
long res = 0;
// res可能超出INT_MAX,比如N=50000,S=1000,假如50个人只会技能1,50人只会技能2,。。。
// 对于人i,有49950个可以教他,总数49950*50000
// 人j被mentor
for(int j = 0; j < N; j++) {
int c = skls[j][0], full = (1 << c) - 1;
int not_mentor = 0;
for(int k = 1; k <= full; k++) {
string key;
for(int m = c-1; m >= 0; m--) {
if(k & (1 << m)) key += to_string(skls[j][c-m]) + " ";
}
if(cnt.count(key)) not_mentor += cnt[key];
}
res += N - not_mentor;
}
cout << res << endl;
}
};
// 人i的技能集是人j的子集,当且仅当i不能教j
int main() {
int T;
cin >> T;
auto sln = new Solution();
for (int i = 1; i <= T; i++)
{
cout << "Case #" << i << ": ";
sln->solve();
}
return 0;
}