hdu5072 数论 + 容斥

分析:

求同色三角形的个数,我们用的是逆转换成求不同色三角形的个数,对于每个顶点,只要和它相连的两个边不同色,它就一定可以组成一个不同色三角形,而且一个三角形会被两次计算。所以一个顶点对不同色三角形的贡献是: prime(n1unprime)/2
问题就转化成了求与一个数不互为质数的数的个数。可以对这个数质因数分解,然后容斥的减去所有含有这些质因数的数的个数,这是一个集合或上的问题显然要容斥。
所以这里预处理了一个数组 divide[i] 表示,能被 i <script type="math/tex" id="MathJax-Element-62">i</script>整除的数的个数,来帮助容斥。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define pr(x) cout << #x << ": " << x << "  " 
#define pl(x) cout << #x << ": " << x << endl;


struct jibancanyang
{
    static const int maxn = int(1e5) + 7;
    int prime[maxn], factor[11];
    bool isprime[maxn];
    int divide[maxn], n, A[maxn], maxNum;

    void getPrime() {
        prime[0] = 0;
        memset(isprime, 0, sizeof(isprime));
        for (int i = 2; i < maxn; ++i) {
            if (!isprime[i]) {
                prime[++prime[0]] = i;
                for (int j = i + i; j < maxn; j += i) isprime[j] = true;
            }
        }
    }

    void getFactor(int x) {
        factor[0] = 0;
        for (int i = 1; i <= prime[0] && prime[i] * prime[i] <= x; ++i) {
            if (x % prime[i] == 0) {
                factor[++factor[0]] = prime[i];
                while (x % prime[i] == 0) x /= prime[i];
            }
        }
        if (x > 1) factor[++factor[0]] = x;
    }

    void getDivide() {
        for (int i = 2; i <= maxNum; ++i) {
            for (int j = i + i; j <= maxNum; j += i) {
                divide[i] += divide[j];
            }
        }
    }


    void fun() {
        int T;
        scanf("%d", &T);
        while (T--) {
            scanf("%d", &n);
            memset(divide, 0, sizeof(divide));
            maxNum = 0;
            for (int i = 0; i < n; ++i) {
                scanf("%d", &A[i]);
                maxNum = max(A[i], maxNum);
                divide[A[i]]++;
            }
            getDivide();
            long long ans = (long long)(n) * (n - 1) * (n - 2) / 6;
            long long unans = 0;
            for (int i = 0; i < n; ++i) {
                getFactor(A[i]);
                long long temp = 0;
                for (int s = 1; s < 1 << factor[0]; ++s) {
                    long long cur = 1;
                    int cnt = 0;
                    for (int j = 0; j < factor[0]; ++j) {
                        if (s >> j & 1) cur *= factor[j + 1], ++cnt;
                    }
                    if (cnt & 1) temp += divide[cur] - 1;
                    else temp -= divide[cur] - 1;
                }
                unans += temp * (n - 1 - temp);
            }
            printf("%lld\n", ans - unans / 2);
        }
    }
}ac;

int main()
{
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
#endif
    ac.getPrime();
    ac.fun();
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值