码迷,mamicode.com
首页 > 其他好文 > 详细

【2017多校训练08 1002】【HDOJ 6134】Battlestation Operational

时间:2017-08-19 23:26:43      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:battle   pen   oid   表达   mat   ++   play   需要   计算   

技术分享

典型的数列反演题。

运用莫比乌斯反演的一个结论 $[n = 1] = \sum_{d | n} \mu(d)$,将表达式做如下转化:

$$ ans = \sum_{i=1}^n \sum_{j=1}^i (\lfloor \frac{i-1}{j} \rfloor + 1) \sum_{d | i \land d | j} \mu(d) \\ = \sum_{d=1}^n \mu(d) \sum_{i=1}^{\lfloor \frac{n}{d} \rfloor} \sum_{j=1}^i (\lfloor \frac{i-1}{j} \rfloor + 1) $$

令$$F_n = \sum_{i=1}^n \sum_{j=1}^i (\lfloor \frac{i-1}{j} \rfloor + 1)$$

则有$$ans = \sum_{d=1}^n \mu(d) F(\lfloor \frac{n}{d} \rfloor)$$

先考虑如何计算Fn.

观察得知,内层求和与n无直接关联,不妨直接对F相邻两项做差:

$$dF_n = F_n - F_{n-1} \\= \sum_{j=1}^n (\lfloor \frac{n-1}{j} \rfloor + 1) $$

考虑每个j对每个n的贡献。

对于一个给定的j,我们可以枚举$\lfloor \frac{n-1}{j} \rfloor$的取值t,此时j和t将对$[j*t+1, j*(t+1)+1)$这一范围内所有的n对应的$F_n$产生t+1的贡献。

由调和级数可知,对所有j枚举它们在N以内的倍数只需要O(Nlog(N))级别的时间复杂度。我们只要在枚举j和t的同时维护一下$dF_n$的相邻两项差,最后做两次前缀和就可以得到$F_n$数列了。

再来考虑如何由$F_n$计算$ans_n$。

由$ans_n = \sum_{d=1}^n \mu(d) F(\lfloor \frac{n}{d} \rfloor)$,与上一步类似,同样可以考虑每个d对每个n的答案的贡献。对于每个d,枚举$\lfloor \frac{n}{d} \rfloor$的取值t,此时d和t对$[t*d, (t+1)*d)$范围内所有的n对应的$ans_n$产生$\mu(d) * F_t$ 的贡献。枚举结束后再做一遍前缀和即可。

技术分享
 1 #include <iostream>
 2 #include <cmath>
 3 #include <cstdio>
 4 #include <algorithm>
 5 using namespace std;
 6 const int maxn = 1000005, mod = 1000000007;
 7 typedef long long LL;
 8 int mu[maxn], muS[maxn], F[maxn], ans[maxn], P[maxn], pcnt, N;
 9 bool not_p[maxn];
10 
11 void sieve()
12 {
13     mu[1] = 1;
14     for(int i = 2;i < maxn;++i)
15     {
16         if(!not_p[i]) P[pcnt++] = i, mu[i] = -1;
17         for(int j = 0;j < pcnt;++j)
18         {
19             if(i * P[j] >= maxn) break;
20             not_p[i * P[j]] = true;
21             if(i % P[j] == 0)
22             {
23                 mu[i * P[j]] = 0;
24                 break;
25             }
26             else mu[i * P[j]] = -mu[i];
27         }
28     }
29 }
30 
31 void init()
32 {
33     sieve();
34     for(int i = 2;i <= N;++i) muS[i] = muS[i-1] + mu[i];
35     N = 1000000;
36     int L, R;
37     for(int k = 1;k < N;++k)
38     {
39         L = k, R = k+k;
40         for(int t = 1;L < N;++t, L += k, R += k)
41         {
42             F[L+1] = (F[L+1] + t) % mod;
43             if(R < N) F[R+1] = (F[R+1] - t) % mod;
44             //if(L < 3) printf("k = %d, (%d, %d] = %d\n", k, L, N, t);
45         }
46     }
47     for(int i = 2;i <= N;++i) F[i] = (F[i] + F[i-1]) % mod;
48     for(int i = 2;i <= N;++i) F[i] = (F[i] + F[i-1]) % mod;
49     for(int i = 1;i <= N;++i)
50         F[i] = (F[i] + (LL) i * (i+1) / 2) % mod;
51     for(int i = 1;i <= N;++i) //F[d]
52     {
53         for(int j = 1, k = i;k <= N;++j, k += i)
54         {
55             int tmp = (mod + F[j] * mu[i]) % mod;
56             ans[k] = (ans[k] + tmp) % mod;
57             if(k+i <= N) ans[k+i] = (ans[k+i] + mod - tmp) % mod;
58         }
59     }
60     for(int i = 2;i <= N;++i) ans[i] = (ans[i] + ans[i-1]) % mod;
61 }
62 
63 void work()
64 {
65     while(~scanf("%d", &N))
66     {
67         printf("%d\n", ans[N]);
68     }
69 }
70 int main() {
71     // your code goes here
72     int T;
73     init();
74     work();
75     return 0;
76 }
莫比乌斯函数,差分数列

 

【2017多校训练08 1002】【HDOJ 6134】Battlestation Operational

标签:battle   pen   oid   表达   mat   ++   play   需要   计算   

原文地址:http://www.cnblogs.com/Asm-Definer/p/7397951.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!