设为首页 加入收藏

TOP

UOJ#310. 【UNR #2】黎明前的巧克力(FWT)
2019-03-16 20:08:14 】 浏览:52
Tags:UOJ#310. UNR 黎明 巧克力 FWT

题意

题目链接

Sol

挂一个讲的看起来比较好的链接

然鹅我最后一步还是没看懂qwq。。

坐等SovietPower大佬发博客

#include<bits/stdc++.h>
using namespace std;
const int MAXN = (1 << 23) + 10, mod = 998244353, inv2 = (mod + 1) / 2, inv4 = 748683265, lim = 1048576;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}
int N, a[MAXN], po3[MAXN];
int add(int x, int y) {
    if(x + y < 0) return x + y + mod;
    return x + y >= mod ? x + y - mod : x + y;
}
int mul(int x, int y) {
    return 1ll * x * y % mod;
}
void FWT(int *a, int opt) {
    for(int mid = 1; mid < lim; mid <<= 1) 
        for(int R = mid << 1, j = 0; j < lim; j += R)
            for(int k = 0; k < mid; k++) {
                int x = a[j + k], y = a[j + k + mid];
                if(opt == 1) a[j + k] = add(x, y), a[j + k + mid] = add(x, -y);
                else a[j + k] = mul(add(x, y), inv2), a[j + k + mid] = mul(add(x, -y), inv2);               
            }
}
int main() {
    N = read();
    for(int i = 1; i <= N; i++) a[read()]++;
    FWT(a, 1);
    po3[0] = 1;
    for(int i = 1; i <= N; i++) po3[i] = mul(3, po3[i - 1]);
    for(int i = 0; i < lim; i++) {
        a[i] = add(mul(2, a[i]), N);
        int c3 = mul(add(a[i], N), inv4);
        a[i] = po3[c3];
        if((N - c3) & 1) a[i] = mod - a[i];
    }
    FWT(a, -1);
    cout << (a[0] - 1 + mod) % mod;
    return 0;
}
】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
上一篇让你的C++代码变的更加健壮 下一篇cf24D. Broken robot(高斯消元)

最新文章

热门文章

Hot 文章

Python

C 语言

C++基础

大数据基础

linux编程基础

C/C++面试题目