BZOJ 3529 [Sdoi2014]数表

神题。。。
假设先不看a的限制,设g(i)为最大公约数为i的数对个数,f(i)为i的约数和,显然有 ans=f(i)g(i) ,而 g(i)=i|dμ(di)ndmd
就有 ans=f(i)i|dμ(di)ndmd
变换一下就有 ans=ndmdi|df(i)μ(di)
接下来就只需要知道 i|df(i)μ(di) 的关于d的前缀和就好了(前缀和怎么做详情可参考popoqqq大神的bzoj2301)
然后线性筛把f(i)和μ(i)搞出来用筛法搞前缀和,如果没有a的限制问题就解决了
那如果有a的限制呢?我们发现只有f(i) <= a的才对答案做出贡献,所以我们把f(i)排序,再把询问按a排序,每次只把小于a的插入前缀和,用树状数组维护前缀和问题就解决了
我不自然溢出就会超时。。可能代码太丑了。。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#define N 100005
using namespace std;
typedef long long ll;
int mu[N], c[N], prime[N], cnt, cas, tot, Ans[N], F[N];
bool no_prime[N];

struct data {
    int n, m, a, id;
    bool operator < (const data &Q)const {
        return a < Q.a;
    }
}ques[N];
struct state {
    int f, id;
    bool operator < (const state &b)const {
        return f < b.f;
    }
}dsum[N];
struct BIT {
    int C[N];
    BIT() { memset(C,0,sizeof(C)); }
    int lowbit(int x) { return x&(-x); }
    void update(int x,int v) { for(int i=x;i<=tot;i+=lowbit(i)) C[i] += v; }
    int query(int x) {
        int ans = 0;
        for(int i=x;i>0;i-=lowbit(i)) ans += C[i];
        return ans;
    }
}t;

void sieve() {
    mu[1] = F[1] = c[1] = 1;
    cnt = 0;
    memset(no_prime,0,sizeof(no_prime));
    for(int i=2;i<=tot;i++) {
        if(!no_prime[i]) {
            prime[++cnt] = i;
            mu[i] = -1;
            F[i] = i+1;
            c[i] = 1;
        }
        for(int j=1, k;(k=prime[j]*i)<=tot;j++) {
            no_prime[k] = 1;
            if(i%prime[j] == 0) {
                mu[k] = 0, c[k] = c[i], F[k] = F[i]*prime[j]+c[i];
                break;
            }
            c[k] = F[i], F[k] = F[i]*(prime[j]+1), mu[k] = -mu[i];
        }
    }
}

template<class T>
inline void read(T &x) {
    x = 0; char op = getchar();
    for(;!isdigit(op);op=getchar());
    for(;isdigit(op);x=x*10+op-'0', op=getchar());
}
char buf[30];
template<class T> 
inline void out(T x) {
    int i=0;
    if(!x)putchar('0');
    while(x) buf[++i] = x%10+'0', x/=10;
    while(i) putchar(buf[i--]);
    putchar('\n');  
} 

int calc(int n,int m) {
    if(n > m) swap(n,m); 
    int ans = 0;
    int last;
    for(int i=1;i<=n;i=last+1) {
        last = min(n/(n/i), m/(m/i));
        ans += (t.query(last)-t.query(i-1))*(n/i)*(m/i);
    }
    return ans&2147483647;
}
void add(state x) {
    int i = x.id;
    for(int j=1;j*i<=tot;j++) t.update(j*i, x.f*mu[j]);
}

int main() {
    scanf("%d",&cas);
    for(int i=1;i<=cas;i++) { read(ques[i].n);read(ques[i].m);read(ques[i].a);ques[i].id = i; tot = max(tot, max(ques[i].n, ques[i].m)); }
    sort(ques+1,ques+1+cas);
    int addcnt = 1;
    sieve();
    for(int i=1;i<=tot;i++) dsum[i] = (state){F[i],i};
    sort(dsum+1,dsum+tot+1);
    for(int i=1;i<=cas;i++) {
        while(dsum[addcnt].f <= ques[i].a && addcnt <= tot) add(dsum[addcnt++]);
        Ans[ques[i].id] = calc(ques[i].n, ques[i].m);
    }   
    for(int i=1;i<=cas;i++) out(Ans[i]);
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值