[CF1007B]Pave the Parallelepiped[组合计数+状态压缩]

题意

\(t\) 组询问，给你 \(A, B, C\) ，问有多少组三元组 \((a, b, c)\) 满足他们任意排列后有： \(a|A,\ b|B,\ c|C\) 。

\(A,B,C,t\leq 10^5\)

分析

• 我们把三个数的所有因子用 \(2^3 - 1\) 个状态表示这个数是 \(A,B,C\) 中的哪几个数字的因子。

• 按照从小到大的顺序枚举3个数对应的集合，首先保证能够找到一种对应方式(每个数对应是谁的因子)，相同的数集利用插板法计算方案避免重复。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cctype>
#include<vector>
#include<queue>
#define For(s) for(int s=1;s<8;s++)
using namespace std;
const int N=1e5 + 7;
typedef long long LL;
int n,T;
int sz[8],A[8],f[8],b[8]={0,1,1,2,1,2,2,3};
LL ans,tmp;
int gcd(int a,int b){
  if(!b) return a;
  return gcd(b,a%b);    
}
int get(int x){
    int res=0;
    for(int i=1;i<=sqrt(x);i++)if(x%i==0){
      res++;
      if(i*i!=x) res++;
    }
    return res;
}
bool check(int a,int b,int c){
    if((a&1) && (b&2) && (c&4)
        return true;
    if((a&1) && (c&2) && (b&4))
        return true;
    if((b&1) && (a&2) && (c&4))
        return true;
    if((b&1) && (c&2) && (a&4))
        return true;
    if((c&1) && (a&2) && (b&4))
        return true;
    if((c&1) && (b&2) && (a&4))
        return true;
    return false;
}
LL C(int n,int m){
    if (m == 0) return 1;
    if (m == 1) return n;
    if (m == 2) return 1ll * n * (n - 1) / 2;
    if (m == 3) return 1ll * n * (n - 1) * (n - 2) / 6;
}
void work(){
     scanf("%d%d%d",&A[0],&A[1],&A[2]);
     ans=0; memset(sz,0,sizeof(sz));
     memset(f,0,sizeof(f));
     For(S){
        for(int i=0;i<3;i++) if(S>>i&1){
            if(!f[S]) f[S]=A[i];
            else f[S]=gcd(f[S],A[i]);
        }
        f[S]=get(f[S]);
     }
     For(s) For(S)if((S&s)==s){
        int cnt=b[S]-b[s];
        sz[s]+=f[S]*(cnt&1?-1:1);
     }
     For(s1)for(int s2=s1;s2<8;s2++)for(int s3=s2;s3<8;s3++){
       if(check(s1,s2,s3)){
          tmp=1;int cnt=1,cho=233;
          if(s1==s2) cho=s1,cnt++;
          if(s2==s3) cho=s2,cnt++;
          if(s1^cho) tmp*=sz[s1];
          if(s2^cho) tmp*=sz[s2];
          if(s3^cho) tmp*=sz[s3];
          if(cnt^1) tmp*=C(sz[cho]+cnt-1,cnt);
          ans+=tmp;
       }
     }
     printf("%lld\n",ans);
}
int main(){
    scanf("%d",&T);
    while(T--) work();
    return 0;
}

转载于:https://www.cnblogs.com/yqgAKIOI/p/10136068.html