牛客 6164 B 牛牛的零食 （容斥原理）

链接： 牛牛的零食

题意：
求给定区间中可以被 8 整除，但不能被 给定数中任何一个数整除的数的个数。

思路：
容斥原理可以求 [l ,r] 能被给定数整除的数的个数，例如 求 能被 2 3 5 整除的数的个数，可以先求出 能被 2 3 5 整除的数的个数，但这时会有重复的出现，所以还要减去，同时被 (2,3),(2,5),(3,5).整除的数的个数，但减的时候右减去了重复的，最后还要加上 同时被(2,3,5) 整除的数的个数。总的来说就是 奇数个数的时候加，偶数个数的时候减。这里可以用二进制表示选了哪些数，然后还要能被 8 整除，所以在计算的时候 每个的初始 lcm 设为 8 .这样每次的结果都是能被 8 整除的前提下算出来的。（也可以不用 二进制，直接 dfs）

代码：

#include<iostream>
#include<cstdio>
#include<map>
#include<math.h>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e6+10;
ll T,a[maxn],k,l,r;
ll gcd(ll a, ll b){
    return b==0 ? a :gcd(b,a%b);
}
ll lcm(ll a,ll b){
    return a * b / gcd (a , b);
}
ll solve(ll n){
    ll  ans=0;
    for(ll i = 1; i < (1 << k) ; i ++){
        ll bits=0,sum=8;
        for(int j = 0; j < k; j++){
            if( (i >> j) & 1){
                bits++; sum = lcm(sum,a[j]);
            }
        }
        if(bits % 2 ==1) ans += n/sum;
        else ans -= n/sum;
    }
    return ans;
}
int main (){
    cin>>k;
    for(int i=0;i<k;i++) cin>>a[i];
    cin>>l>>r;
    ll ans=r/8-(l-1)/8;
    cout<<ans-solve(r)+solve(l-1)<<endl;
}

dfs 代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn=2e3+7;
typedef long long ll;
ll a[20],n,x,y,ans;
ll gcd(ll a, ll b){
    if(b==0) return a;
    return gcd(b,a%b);
}
ll lcm(ll a,ll b){
    return a*b/gcd(a,b);
}
void dfs(ll pos,ll last,ll num){
    if(pos%2==0) ans+=y/num-x/num;
    else ans-=y/num-x/num;
    for(int i=last+1;i<=n;i++){
         dfs(pos+1,i,lcm(num,a[i]));
     }
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    cin>>x>>y;
    x--;
    dfs(0,0,8);
    printf ("%lld\n",ans);
 
}