容斥原理自学笔记

最新推荐文章于 2023-06-08 17:25:35 发布

liyuu123

最新推荐文章于 2023-06-08 17:25:35 发布

阅读量250

点赞数

分类专栏：笔记容斥原理文章标签： C++

本文链接：https://blog.csdn.net/qq_37322063/article/details/78405214

版权

笔记同时被 2 个专栏收录

2 篇文章 0 订阅

订阅专栏

容斥原理

1 篇文章 0 订阅

订阅专栏

在考虑“情况的筛选”的过程中可能重复考虑了同种情况，需要将多余其排除，但也可能过度排除（多次排除同种情况），则还需要将“误伤”情况重新加回来，如此反复，直到没有多余操作为止，这就是容斥原理的原理（笑）

以一个例题切入：

heal

有一条长轴，坐标 0 为起点，
已知将经历N次操作，每次操作会使坐标移动X的整数倍（包括 0 倍）
已知操作次数及最后一次操作的X值，之前所有操作的值都不大于最后一次的值
问有多少种可行的操作的值组合，使坐标最终到达 1？

分析

也就是在前N个数中，至少要有一个数与X的gcd为1，由此推出，可行的情况为总的情况减去N个数各个质因数且不互质的情况，考虑到可能重复删减，需要用容斥原理解决

容斥原理通常有两种实现方法：打表法& $dfs$ （递推/归）法，以下两个模板：

容斥原理模板（heal）

//打表模板
#include<bits/stdc++.h>
#define mod 1000000007
#define ll long long
using namespace std;
ll read()
{
    ll x=0,f=1;char c;
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-1;
    for(;isdigit(c);c=getchar())x=(x<<3)+(x<<1)+c-'0';
    return x*f;
}
ll n,m;
int tot=0;
ll a[2005];
ll ans;
void calc(ll x){
    for(ll i=2;i*i<=m;i++){
        if(x%i==0){
            a[++tot]=i;
            while(x%i==0)x/=i;
        }
    }
    if(x!=1)a[++tot]=x;
}
ll ksm(ll a,ll b){
    ll ans=1;
    while(b){
        if(b&1)ans=(ans*a)%mod;
        a=(a*a)%mod;
        b>>=1;
    }
    return ans;
}
int main()
{
    n=read();m=read();
    calc(m);
    for(ll i=1;i<=(1<<tot);i++){
        ll k=1,tmp=1;
        for(ll j=1;j<=tot;j++)
            if(i&(1<<(j-1))){
                k=-k;tmp*=a[j];
            }
        ans+=(ll)ksm(m/tmp%mod,n)*k;
        ans=(ans%mod+mod)%mod;
    }
    cout<<ans;
}

//dfs模板
#include<bits/stdc++.h>
using namespace std;
const long long p = 1e9+7;
long long n,m,ans;
long long fc[100];

inline void Div(long long x){
    for(long long i=2;i*i<=x;i++)
      if(x%i==0){
         fc[++fc[0]]=i;
         while(x%i==0) x/=i;
      }
    if(x!=1) fc[++fc[0]]=x;
}

inline long long ksm(long long a,long long b){
    long long ret=1;
    a%=p;
    for(;b;b>>=1,a=(1ll*a*a)%p)
      if(b&1) ret=(ret*a)%p;
    return ret;
}

void dfs(int x,long long d,int f)
{
    if(x==fc[0]+1)
    {
        ans+=f*ksm(m/d,n);
        return;
    }
    dfs(x+1,d*fc[x],f*-1);
    dfs(x+1,d,f);
}

int main(){
//  freopen("lx.in","r",stdin);
    scanf("%lld%lld",&n,&m);
    Div(m);
    dfs(1,1,1);
    cout<<(ans%p+p)%p;
}