Kick Start 2019 F Teach Me

文章目录

链接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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值