数论 - 容斥原理

Calebbbbb

已于 2024-02-20 11:28:37 修改

阅读量382

点赞数 2

分类专栏：算法模板文章标签：算法 c++ c语言容斥原理

于 2024-02-20 11:28:17 首次发布

本文链接：https://blog.csdn.net/2203_75327677/article/details/136186408

版权

算法模板专栏收录该内容

21 篇文章

订阅专栏

文章目录

一、题目描述
二、算法思路
三、代码

在计数时，必须注意没有重复，没有遗漏。为了使重叠部分不被重复计算，人们研究出一种新的计数方法，这种方法的基本思想是：先不考虑重叠的情况，把包含于某内容中的所有对象的数目先计算出来，然后再把计数时重复计算的数目排斥出去，使得计算的结果既无遗漏又无重复，这种计数的方法称为容斥原理。

一、题目描述

能被整除的数

给定一个整数 n 和 m 个不同的质数 p1,p2,…,pm。

请你求出 1∼n1 中能被 p1,p2,…,pm 中的至少一个数整除的整数有多少个。

输入格式

第一行包含整数 n 和 m。

第二行包含 m 个质数。

输出格式

输出一个整数，表示满足条件的整数的个数。

数据范围

1≤m≤16,
1≤n,pi≤10⁹

输入样例：

10 2
2 3

输出样例：

二、算法思路

容斥原理公式：
用二进制表示：上面容斥原理公式右边一共有2^m-1项。所以我们可以用二进制表示每个约数是否存在，比如1则表示约数只有p1，1001则表示约数时p4*p1
通过一共2^m-1项进行二进制遍历，计算出公式答案
细节在代码中阐释

三、代码

#include <iostream>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 20;

int p[N];

int main()
{
    int n, m;
    cin >> n >> m;

    for (int i = 0; i < m; i ++ ) cin >> p[i];

    int res = 0; //答案
    for (int i = 1; i < 1 << m; i ++ ) //1 << m是2的m次方
    {
        int t = 1, s = 0; //t表示该二进制选择情况的约数，s表示该约数有多少个1
        for (int j = 0; j < m; j ++ )
            if (i >> j & 1)
            {
                if ((LL)t * p[j] > n) //约数大于被约束不成立
                {
                    t = -1;
                    break;
                }
                t *= p[j];
                s ++ ;
            }

        if (t != -1)
        {
	        //通过1个数来判断正负号
            if (s % 2) res += n / t;
            else res -= n / t;
        }
    }

    cout << res << endl;

    return 0;
}