100% itl memset blog read 線段 太陽 break als

棟棟有一塊長方形的地，他在地上種了一種能量植物，這種植物可以采集太陽光的能量。在這些植物采集能量後，

棟棟再使用一個能量匯集機器把這些植物采集到的能量匯集到一起。 棟棟的植物種得非常整齊，一共有n列，每列 有m棵，植物的橫豎間距都一樣，因此對於每一棵植物，棟棟可以用一個坐標(x, y)來表示，其中x的範圍是1至n， 表示是在第x列，y的範圍是1至m，表示是在第x列的第y棵。 由於能量匯集機器較大，不便移動，棟棟將它放在了 一個角上，坐標正好是(0, 0)。 能量匯集機器在匯集的過程中有一定的能量損失。如果一棵植物與能量匯集機器 連接而成的線段上有k棵植物，則能量的損失為2k + 1。例如，當能量匯集機器收集坐標為(2, 4)的植物時，由於 連接線段上存在一棵植物(1, 2)，會產生3的能量損失。註意，如果一棵植物與能量匯集機器連接的線段上沒有植 物，則能量損失為1。現在要計算總的能量損失。 下面給出了一個能量采集的例子，其中n = 5，m = 4，一共有20 棵植物，在每棵植物上標明了能量匯集機器收集它的能量時產生的能量損失。 在這個例子中，總共產生了36的能 量損失。

Input

僅包含一行，為兩個整數n和m。

Output

僅包含一個整數，表示總共產生的能量損失。

Sample Input

【樣例輸入1】
5 4
【樣例輸入2】
3 4

Sample Output

【樣例輸出1】
36
【樣例輸出2】
20
對於100%的數據：1 ≤ n, m ≤ 100,000。

---

　　題目大意 求

　　因為n和m單個比較小，而且gcd(i, j)不超過min(n, m)，因此可以想到去枚舉gcd(i, j)的取值，然後統計個數，再乘一乘即可。

　　至於這個統計個數有一個很好的方法就是容斥原理，計算i和j都是d的倍數時的對數，這個很好算，直接搬結論吧，是

，然後再減去gcd(i, j)的值為2d, 3d, ...的時候。

　　這樣顯然倒著算能夠更簡單，這樣的時間復雜度是O(nlog₂n)。

Code

 1 /**
 2  * bzoj
 3  * Problem#2005
 4  * Accepted
 5  * Time:24ms
 6  * Memory:2068k
 7  */
 8 #include <iostream>
 9 #include <cstdio>
10 #include <ctime>
11 #include <cmath>
12 #include <cctype>
13 #include <cstring>
14 #include <cstdlib>
15 #include <fstream>
16 #include <sstream>
17 #include <algorithm>
18 #include <map>
19 #include <set>
20 #include <stack>
21 #include <queue>
22 #include <vector>
23 #include <stack>
24 #ifndef WIN32
25 #define Auto "%lld"
26 #else
27 #define Auto "%I64d"
28 #endif
29 using namespace std;
30 typedef bool boolean;
31 const signed int inf = (signed)((1u << 31) - 1);
32 const signed long long llf = (signed long long)((1ull << 61) - 1);
33 const double eps = 1e-6;
34 const int binary_limit = 128;
35 #define smin(a, b) a = min(a, b)
36 #define smax(a, b) a = max(a, b)
37 #define max3(a, b, c) max(a, max(b, c))
38 #define min3(a, b, c) min(a, min(b, c))
39 template<typename T>
40 inline boolean readInteger(T& u){
41     char x;
42     int aFlag = 1;
43     while(!isdigit((x = getchar())) && x != ‘-‘ && x != -1);
44     if(x == -1) {
45         ungetc(x, stdin);    
46         return false;
47     }
48     if(x == ‘-‘){
49         x = getchar();
50         aFlag = -1;
51     }
52     for(u = x - ‘0‘; isdigit((x = getchar())); u = (u << 1) + (u << 3) + x - ‘0‘);
53     ungetc(x, stdin);
54     u *= aFlag;
55     return true;
56 }
57 
58 int n, m;
59 
60 inline void init() {
61     readInteger(n);
62     readInteger(m);
63     if(n > m)    swap(n, m);
64 }
65 
66 long long res = 0;
67 long long f[100005];
68 inline void solve() {
69     for(int d = n; d; d--) {
70         f[d] = (n / d) * 1LL * (m / d);
71         for(int i = (d << 1); i <= n; i += d)
72             f[d] -= f[i];
73         res += d * 1LL * f[d];
74     }
75     res = (res << 1) - m * 1LL * n;
76     printf(Auto, res);
77 }
78 
79 int main() {
80     init();
81     solve();
82     return 0;
83 }
 

Solution 1

　　但是另外用點黑科技，可以讓它變成O(n)。

　　記得有一個關於歐拉函數的結論。於是有:

　　

　　現在等同於求有多少個數對(i, j)使得i能夠整除d，j能夠整除d，這是一個剛剛就解決了的問題，於是我們得到了下面這個優美的式子:

　　對於求phi函數的值這個事情就交給已經閑置了一會兒的線性篩去做吧。

Code

  1 /**
  2  * bzoj
  3  * Problem#2005
  4  * Accepted
  5  * Time:4ms
  6  * Memory:2172k
  7  */
  8 #include <iostream>
  9 #include <cstdio>
 10 #include <ctime>
 11 #include <cmath>
 12 #include <cctype>
 13 #include <cstring>
 14 #include <cstdlib>
 15 #include <fstream>
 16 #include <sstream>
 17 #include <algorithm>
 18 #include <map>
 19 #include <set>
 20 #include <stack>
 21 #include <queue>
 22 #include <vector>
 23 #include <list>
 24 #ifndef WIN32
 25 #define Auto "%lld"
 26 #else
 27 #define Auto "%I64d"
 28 #endif
 29 using namespace std;
 30 typedef bool boolean;
 31 const signed int inf = (signed)((1u << 31) - 1);
 32 const signed long long llf = (signed long long)((1ull << 61) - 1);
 33 const double eps = 1e-6;
 34 const int binary_limit = 128;
 35 #define smin(a, b) a = min(a, b)
 36 #define smax(a, b) a = max(a, b)
 37 #define max3(a, b, c) max(a, max(b, c))
 38 #define min3(a, b, c) min(a, min(b, c))
 39 template<typename T>
 40 inline boolean readInteger(T& u){
 41     char x;
 42     int aFlag = 1;
 43     while(!isdigit((x = getchar())) && x != ‘-‘ && x != -1);
 44     if(x == -1) {
 45         ungetc(x, stdin);    
 46         return false;
 47     }
 48     if(x == ‘-‘){
 49         x = getchar();
 50         aFlag = -1;
 51     }
 52     for(u = x - ‘0‘; isdigit((x = getchar())); u = (u << 1) + (u << 3) + x - ‘0‘);
 53     ungetc(x, stdin);
 54     u *= aFlag;
 55     return true;
 56 }
 57 
 58 const int limit = 1e5;
 59 
 60 int n, m;
 61 int num = 0;
 62 int prime[limit + 1];
 63 int phi[limit + 1];
 64 boolean vis[limit + 1];
 65 long long res = 0;
 66 
 67 inline void init() {
 68     readInteger(n);
 69     readInteger(m);
 70     if(n > m)    swap(n, m);
 71 }
 72 
 73 inline void Euler() {
 74     phi[1] = 1;
 75     memset(vis, false, sizeof(boolean) * (n + 1));
 76     for(int i = 2; i <= n; i++) {
 77         if(!vis[i])    prime[num++] = i, phi[i] = i - 1;
 78         for(int j = 0; j < num && i * 1LL * prime[j] <= n; j++) {
 79             int c = i * prime[j];
 80             vis[c] = true;
 81             if((i % prime[j]) == 0) {
 82                 phi[c] = prime[j] * phi[i];
 83                 break;
 84             } else {
 85                 phi[c] = phi[prime[j]] * phi[i];
 86             }
 87         }
 88     }
 89 }
 90 
 91 inline void solve() {
 92     for(int i = 1; i <= n; i++)
 93         res += (n / i) * 1LL * (m / i) * phi[i];
 94     printf(Auto"\n", (res * 2) - (long long)n * m);
 95 }
 96 
 97 int main() {
 98     init();
 99     Euler();
100     solve();
101     return 0;
102 }

bzoj 2005 能量采集 - 容斥原理