标签: 数学

[SDOI2014]数表 题解

[SDOI2014]数表 题解

题目地址:洛谷:【P3312】[SDOI2014]数表 – 洛谷、BZOJ:Problem 3529. — [Sdoi2014]数表

题目描述

有一张N*m的数表,其第i行第j列(1 < =i < =n,1 < =j < =m)的数值为能同时整除i和j的所有自然数之和。给定a,计算数表中不大于a的数之和。

输入输出格式

输入格式:
输入包含多组数据。
输入的第一行一个整数Q表示测试点内的数据组数
接下来Q行,每行三个整数n,m,a(|a| < =10^9)描述一组数据。

输出格式:
对每组数据,输出一行一个整数,表示答案模2^31的值。

输入输出样例

输入样例#1:

2
4 4 3
10 10 5

输出样例#1:

20
148

说明

1 < =N.m < =10^5 , 1 < =Q < =2*10^4

题解

参考资料:题解 P3312 【[SDOI2014]数表】 – ldlsn 的博客 – 洛谷博客
神奇的反演。
冷静分析一下,如果固定i和j,如何求能同时整除i和j的所有自然数之和。规定F(x)为x的因数和,例如F(2)=1+2=3, F(6)=1+2+3+6=12,则上面这个问题的答案就是F[\mathrm{gcd}(i, j)]
得到了上面的结论后,不考虑a的限制,我们可以知道答案就是\sum_{i=1}^n \sum_{j=1}^m F[\mathrm{gcd}(i, j)],下面就是对这个答案式的变形。
我们知道常用的反演结论g(x) = \sum_{i=1}^n \sum_{j=1}^m [\mathrm{gcd}(i, j) == x] = \sum_{i|d} \mu \left( \frac{d}{i} \right) \left\lfloor \frac{n}{d} \right\rfloor \left\lfloor \frac{m}{d} \right\rfloor,直接套进答案里
\sum_{i=1}^n F(i)g(i) = \sum_{i=1}^n F(i) \sum_{i|d} \mu \left( \frac{d}{i} \right) \left\lfloor \frac{n}{d} \right\rfloor \left\lfloor \frac{m}{d} \right\rfloor
改变一下求和顺序
\sum_{d=1}^n \left\lfloor \frac{n}{d} \right\rfloor \left\lfloor \frac{m}{d} \right\rfloor \sum_{i|d} F(i) \mu \left( \frac{d}{i} \right)
如果我们可以计算出f(x) = \sum_{i|x} F(i) \mu(\frac{x}{i})的值前缀和,那么可以除法分块做,而F(x)可以枚举倍数求和,这些都不是问题。
接下来我们来看带上限制应该怎么做,考虑离线处理,把询问按a限制从小到大排序,每一次将符合a条件的F(x)f(x)产生的贡献扔进一个维护前缀和的数据结构中(比如树状数组),为回答询问的除法分块求前缀和做准备。
至于那个奇怪的模数,自然溢出后&0x7fffffff就好。
总复杂度是O(n \sqrt{n} \log n)

代码

// Code by KSkun, 2018/6
#include <cstdio>
#include <cctype>

#include <algorithm>

typedef long long LL;

inline char fgc() {
    static char buf[100000], *p1 = buf, *p2 = buf;
    return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 100000, stdin), p1 == p2) 
        ? EOF : *p1++;
}

inline LL readint() {
    register LL res = 0, neg = 1;
    register char c = fgc();
    while(!isdigit(c)) {
        if(c == '-') neg = -1;
        c = fgc();
    }
    while(isdigit(c)) {
        res = (res << 1) + (res << 3) + c - '0';
        c = fgc();
    }
    return res * neg;
}

const int MAXN = 100005;

int prime[MAXN], mu[MAXN], ptot;
bool notprime[MAXN];

inline void sieve() {
    notprime[1] = true; mu[1] = 1;
    for(int i = 2; i < MAXN; i++) {
        if(!notprime[i]) {
            prime[++ptot] = i;
            mu[i] = -1;
        }
        for(int j = 1; j <= ptot && i * prime[j] < MAXN; j++) {
            int t = i * prime[j];
            notprime[t] = true;
            if(i % prime[j]) {
                mu[t] = -mu[i];
            } else {
                mu[t] = 0; break;
            }
        }
    }
}

int bit[MAXN];

inline int lowbit(int x) {
    return x & -x;
}

inline void add(int x, int v) {
    for(int i = x; i < MAXN; i += lowbit(i)) {
        bit[i] += v;
    }
}

inline int query(int x) {
    int res = 0;
    for(int i = x; i; i -= lowbit(i)) {
        res += bit[i];
    }
    return res;
}

struct Node {
    int x, v;
    inline bool operator<(const Node &rhs) const {
        return v < rhs.v;
    }
} F[MAXN];

inline void calf() {
    for(int i = 1; i <= 100000; i++) {
        F[i].x = i;
        for(int j = 1; i * j <= 100000; j++) {
            F[i * j].v += i;
        }
    }
    std::sort(F + 1, F + 100001);
}

inline int calc(int n, int m) {
    int ans = 0;
    if(n > m) std::swap(n, m);
    for(int i = 1; i <= n;) {
        int j = std::min(n / (n / i), m / (m / i));
        ans += (query(j) - query(i - 1)) * (n / i) * (m / i);
        i = j + 1;
    }
    return ans;
}

struct Query {
    int n, m, a, id;
} qur[MAXN];

inline bool qcmp(Query a, Query b) {
    return a.a < b.a;
}

int T, ans[MAXN];

int main() {
    sieve(); calf();
    T = readint();
    for(int i = 1; i <= T; i++) {
        qur[i].n = readint(); qur[i].m = readint(); qur[i].a = readint(); qur[i].id = i;
    }
    std::sort(qur + 1, qur + T + 1, qcmp);
    int now = 0;
    for(int i = 1; i <= T; i++) {
        while(now < 100000 && F[now + 1].v <= qur[i].a) {
            now++;
            for(int j = 1; j * F[now].x <= 100000; j++) {
                add(j * F[now].x, mu[j] * F[now].v);
            }
        }
        ans[qur[i].id] = calc(qur[i].n, qur[i].m);
    }
    for(int i = 1; i <= T; i++) {
        printf("%d\n", ans[i] & 0x7fffffff);
    }
    return 0;
}
[HAOI2011]Problem b 题解

[HAOI2011]Problem b 题解

题目地址:洛谷:【P2522】[HAOI2011]Problem b – 洛谷、BZOJ:Problem 2301. — [HAOI2011]Problem b

题目描述

对于给出的n个询问,每次求有多少个数对(x,y),满足a≤x≤b,c≤y≤d,且gcd(x,y) = k,gcd(x,y)函数为x和y的最大公约数。

输入输出格式

输入格式:
第一行一个整数n,接下来n行每行五个整数,分别表示a、b、c、d、k

输出格式:
共n行,每行一个整数表示满足要求的数对(x,y)的个数

输入输出样例

输入样例#1:

2
2 5 1 5 1
1 5 1 5 2

输出样例#1:

14
3

说明

100%的数据满足:1≤n≤50000,1≤a≤b≤50000,1≤c≤d≤50000,1≤k≤50000

题解

这个题[POI2007]ZAP-QUERIES的一个强化版。我们考虑使用容斥原理,答案即ans([1, b], [1, d])-ans([1, a-1], [1, d])-ans([1, b], [1, c-1])+ans([1, a-1], [1, c-1])。
总复杂度O(n \sqrt{n})

代码

// Code by KSkun, 2018/6
#include <cstdio>
#include <cctype>  

#include <algorithm>

typedef long long LL;

inline char fgc() {
    static char buf[100000], *p1 = buf, *p2 = buf;
    return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 100000, stdin), p1 == p2) 
        ? EOF : *p1++;
}

inline LL readint() {
    register LL res = 0, neg = 1;
    register char c = fgc();
    while(!isdigit(c)) {
        if(c == '-') neg = -1;
        c = fgc();
    }
    while(isdigit(c)) {
        res = (res << 1) + (res << 3) + c - '0';
        c = fgc();
    }
    return res * neg;
}

const int MAXN = 50005;

bool notprime[MAXN];
int prime[MAXN], miu[MAXN], ptot;

inline void sieve() {
    miu[1] = 1;
    for(int i = 2; i < MAXN; i++) {
        if(!notprime[i]) {
            prime[ptot++] = i;
            miu[i] = -1;
        }
        for(int j = 0; j < ptot && prime[j] * i < MAXN; j++) {
            int k = prime[j] * i;
            notprime[k] = true;
            if(i % prime[j] == 0) {
                miu[k] = 0; break;
            } else {
                miu[k] = -miu[i];
            }
        }
    }
    for(int i = 2; i < MAXN; i++) {
        miu[i] += miu[i - 1];
    }
}

inline LL cal(int b, int d, int k) {
    LL all = 0;
    if(!k) return 0;
    b /= k; d /= k;
    if(b > d) std::swap(b, d);
    for(int i = 1; i <= b;) {
        int j = std::min(b / (b / i), d / (d / i));
        all += 1ll * (miu[j] - miu[i - 1]) * (b / i) * (d / i);
        i = j + 1;
    }
    return all;
}

int T, a, b, c, d, k;

int main() {
    sieve();
    T = readint();
    while(T--) {
        a = readint(); b = readint(); c = readint(); d = readint(); k = readint();
        printf("%lld\n", cal(b, d, k) - cal(a - 1, d, k) - cal(b, c - 1, k) 
            + cal(a - 1, c - 1, k));
    }
    return 0;
}
[ZJOI2010]排列计数 题解

[ZJOI2010]排列计数 题解

题目地址:洛谷:【P2606】[ZJOI2010]排列计数 – 洛谷、BZOJ:Problem 2111. — [ZJOI2010]Perm 排列计数

题目描述

称一个1,2,…,N的排列P1,P2…,Pn是Magic的,当且仅当2<=i<=N时,Pi>Pi/2. 计算1,2,…N的排列中有多少是Magic的,答案可能很大,只能输出模P以后的值

输入输出格式

输入格式:
输入文件的第一行包含两个整数 n和p,含义如上所述。

输出格式:
输出文件中仅包含一个整数,表示计算1,2,⋯, ���的排列中, Magic排列的个数模 p的值。

输入输出样例

输入样例#1:

20 23 

输出样例#1:

16

说明

100%的数据中,1 ≤N ≤ 10^6, P≤ 10^9,p是一个质数。

题解

发现了一个奇特的性质,如果把这个数组看成是一棵二叉树,节点i对应的左右儿子分别是2i和2i+1,那么这个序列的二叉树是一个小根堆。
我们考虑一个分治的思想,对于一个数集建立完全二叉树的小根堆,一定是取最小元素当堆顶,然后将元素分成左右子树再进行计算,由于左右子树的大小一定比当前数集的大小小,可以使用动态规划来解决这个问题。我们从数集中选出左子树那么多元素还要乘上一个组合系数,即:
dp[i] = \mathrm{C}_{i-1}^l dp[l] dp[r]
由于我们不能O(n^2)地预处理组合数,必须使用Lucas定理来求。划分左右子树的时候,可以先检查左子树是否填满,满了就往右子树填。要判断左子树满没满,我们可以计算出最后一层有n - (2^{\lfloor \log n \rfloor} - 1)这么多节点,判断最后一层满一半没有即可。

代码

// Code by KSkun, 2018/5
#include <cstdio>
#include <cctype>

#include <algorithm>

typedef long long LL;

inline char fgc() {
    static char buf[100000], *p1 = buf, *p2 = buf;
    return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 100000, stdin), p1 == p2) ? EOF : *p1++;
}

inline LL readint() {
    register LL res = 0, neg = 1;
    register char c = fgc();
    while(!isdigit(c)) {
        if(c == '-') neg = -1;
        c = fgc();
    }
    while(isdigit(c)) {
        res = (res << 1) + (res << 3) + c - '0';
        c = fgc();
    }
    return res * neg;
}

const int MAXN = 1000005;

LL n, p, Log[MAXN], mul[MAXN], inv[MAXN], dp[MAXN];

inline void init() {
    Log[2] = 1;
    for(int i = 3; i < MAXN; i++) {
        Log[i] = Log[i >> 1] + 1;
    }
    mul[0] = mul[1] = inv[0] = inv[1] = 1;
    for(int i = 2; i < MAXN; i++) {
        inv[i] = (-(p / i) * inv[p % i] % p + p) % p;
    }
    for(int i = 2; i < MAXN; i++) {
        mul[i] = i * mul[i - 1] % p;
        inv[i] = inv[i] * inv[i - 1] % p;
    }
}

inline LL lucas(LL n, LL m) {
    if(m > n) return 0;
    if(n < p && m < p) return mul[n] * inv[n - m] % p * inv[m] % p;
    return lucas(n / p, m / p) * lucas(n % p, m % p) % p;
}

int main() {
    n = readint(); p = readint();
    init();
    dp[1] = dp[2] = 1 % p; dp[3] = 2 % p;
    int l = 1, r = 1;
    for(int i = 4; i <= n; i++) {
        if(i - (1 << Log[i]) + 1 <= 1 << (Log[i] - 1)) l++; else r++;
        dp[i] = lucas(i - 1, l) * dp[l] % p * dp[r] % p;
    }
    printf("%lld", dp[n]);
    return 0;
}