hdu 5514 Frogs(计数)

题目链接:hdu 5514 Frogs

解题思路

首先,一只青蛙的步长为x,那么k * gcd(x,m)都是这只青蛙会经过的地方。
如果直接计算每只青蛙的贡献,会有重复的地方被计算多次。

我的做法是将m的因子处理出来,判断每个因子的d,是否k*d的地方会被经过。然后从小的因子开始,如果经过,但未加上贡献,就加上它的贡献,并且更新比它大的因子;如果经过多次,则要减掉超出的贡献。

代码

#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <algorithm>

using namespace std;
typedef long long ll;
const int maxn = 1e3 + 5;

int N, M, F[maxn], C[maxn];
vector<int> G;

void destory(int n) {
    int m = (int)sqrt(n);
    G.clear();

    for (int i = 1; i <= m; i++) {
        if (n % i == 0) {
            G.push_back(i);
            if (n / i != i)
                G.push_back(n / i);
        }
    }
    sort(G.begin(), G.end());
}

int gcd(int a, int b) { return b == 0 ? a : gcd(b, a % b); }
int find(int x) { return lower_bound(G.begin(), G.end(), x) - G.begin(); }
ll get(int x, int s) { return 1LL * x * (x + 1) / 2 * s; }

ll solve () {
    memset(F, 0, sizeof(F));
    memset(C, 0, sizeof(C));

    int u;
    for (int i = 1; i <= N; i++) {
        scanf("%d", &u);
        int d = gcd(u, M);
        F[find(d)] = 1;
    }

    int n = G.size();
    for (int i = 0; i < n; i++) {
        if (!F[i]) continue;
        for (int j = i + 1; j < n; j++)
            if (F[j] == 0 && G[j] % G[i] == 0) F[j] = 1;
    }

    ll ans = 0;
    for (int i = 0; i < n; i++) {
        int v = F[i] - C[i];

        ans = ans + get(M/G[i]-1, G[i]) * v;
        for (int j = i; j < n; j++)
            if (G[j] % G[i] == 0) C[j] += v;
    }
    return ans;
}

int main () {
    int cas;
    scanf("%d", &cas);
    for (int kcas = 1; kcas <= cas; kcas++) {
        scanf("%d%d", &N, &M);
        destory(M);
        printf("Case #%d: %lld\n", kcas, solve());
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值