2022年牛客多校第四场补题

D.Jobs (Easy Version)

题意

$q$个人找工作，有$n$家公司，每家公司有$m$个工作，每个人和每一份工作都有$3$个属性$a，b，c$，当且仅当每个人的$a，b，c$都大于某一份工作的$a，b，c$，这个人才能成功完成这个工作，如果一个人能完成某公司的一项工作，那么这个公司就会要他。每个人的属性都是由给你的$seed$作为随机数的种子和上一个人的答案，进行随机生成。

计算

$({\sum }_{i=1}^{n}an{s}_i\ast see{d}^{q-i})mod998244353$

$an{s}_i$表示第i个人有多少公司愿意接纳他

数据范围

$1⩽n⩽10$

$1⩽q⩽2\ast 10^6$

$1⩽m⩽10^5$

$1⩽a_i,b_i,c_i⩽400$

思路

由于有$q$个人了，那么查询一个人的答案的时间复杂度就要在$\log (m\ast n)$，能想到先预处理出对于公司$i$某个能力的值能否胜任， 然后$O(1)$进行查询。

预处理，开四维的$bool$数组会爆内存，所以用$bitset$进行压位，$dp$一下：$dp[i][a][b][c]$一个人的能力属性值为$a,b,c$能否胜任公司$i$。

时间复杂度

$O(n\ast 400^3)$

可以优化的地方

• 1.可以优化成三维的

• 2.语言选择 $C++(g++\mathrm{7.5.0})$，选$C++(clang++\mathrm{11.0.1})$就超时了。

• 3.中间的快速幂可以先预处理出来可以少个$log$。

代码：

#include 
#pragma gcc optimize("O2")
#pragma g++ optimize("O2")
//#define int long long
#define endl '\n'
using namespace std;


const int N = 2e5 + 10, MOD = 998244353;
int n, q, x, y, z;
int k1,k2,k3,k4;
bitset<402>fk[11][401][401];
inline int getid(int x,int y,int z){
    return 160801*(x)+(y)*(401)+z;
}
int getans(int iq, int eq, int aq){
    int ans=0;
    for(int i=1;i<=n;i++){
        ans+=fk[i][iq][eq][aq];
    }
    return ans;
}

int binpow(long long x, int y, int mod, long long res = 1){
    for (; y; y >>= 1, (x *= x) %= mod) if (y & 1) (res *= x) %= mod;
    return res;
}

int powq[2000010];

inline void solve(){
    cin >> n >> q;
    for(int i = 1; i <= n; i++){
        cin >> k1;
        for(int j = 1; j <= k1; j++) {
            cin>>x>>y>>z;
            fk[i][x][y][z]=1;
        }
    } 
    
    for(short i=1;i<=n;i++){
        for(short j=1;j<=400;j++){
            for(short k=1;k<=400;k++){
                for(short x=1;x<=400;x++){
                    fk[i][j][k][x]=fk[i][j][k][x]|fk[i][j][k][x-1]|fk[i][j][k-1][x]|fk[i][j-1][k][x];
                }
            }
        }
    }
    int seed; cin >> seed;
    powq[0] = 1;
    for(int i = 1; i <= 2000000 + 5; i++){
        powq[i] = 1ll * powq[i - 1] * seed % MOD;
    }
    std::mt19937 rng(seed);
    std::uniform_int_distribution<> u(1, 400);
    int lastans = 0, ans = 0;
    for (int i = 1; i <= q; i++){
        int IQ = (u(rng) ^ lastans) % 400 + 1;  // The IQ of the i-th friend
        int EQ = (u(rng) ^ lastans) % 400 + 1;  // The EQ of the i-th friend
        int AQ = (u(rng) ^ lastans) % 400 + 1;  // The AQ of the i-th friend
        lastans = getans(IQ, EQ, AQ);           // The answer to the i-th friends
        (ans += (1ll * lastans * powq[q - i] % MOD)) %= MOD;
    }
    cout << (ans % MOD) << endl;
}

signed main(){
    
    ios_base::sync_with_stdio(false), cin.tie(0);
    cout << fixed << setprecision(12);
    int t = 1;
    while(t--) solve();
    return 0;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75