1. 程式人生 > 實用技巧 >2018ICPC 南京Problem J. Prime Game

2018ICPC 南京Problem J. Prime Game

題目:

題意:給出一個序列a1,,ana1,⋯,an。fac(l,r)fac(l,r)為mul(l,r)mul(l,r)中不同質因數的個數。

請計算:

i=1nj=infacl,r 思路:求質因數的貢獻度,我們可以定義一個二維vector pos[i][k]=p表示當前質因數i在p位置出現k次那麼該因子的貢獻度為(n-p+1)*p, 因為可能會重複計算的,那麼我們只要減去上一個位置出現i的情況,那麼每一個質因子的貢獻度為(n-pos[i][k]+1)*(pos[i][k]-pos[i][k-1]; 有兩種實現一種直接質數分解,大概跑了800ms,用一個尤拉篩進行打表優化跑了400ms。
  1 //#include<bits/stdc++.h>
  2 #include<time.h>
  3 #include <set>
  4 #include <map>
  5 #include <stack>
  6 #include <cmath>
  7 #include <queue>
  8 #include <cstdio>
  9 #include <cstring>
 10 #include <string>
 11 #include <vector>
 12
#include <cstring> 13 #include <iostream> 14 #include <algorithm> 15 #include <list> 16 using namespace std; 17 #define eps 1e-10 18 #define PI acos(-1.0) 19 #define lowbit(x) ((x)&(-x)) 20 #define zero(x) (((x)>0?(x):-(x))<eps) 21 #define mem(s,n) memset(s,n,sizeof s); 22
#define ios {ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);} 23 typedef long long ll; 24 typedef unsigned long long ull; 25 const int maxn=1e6+5; 26 const int Inf=0x7f7f7f7f; 27 const ll Mod=1e9+7; 28 const int N=3e3+5; 29 bool isPowerOfTwo(int n) { return n > 0 && (n & (n - 1)) == 0; }//判斷一個數是不是 2 的正整數次冪 30 int modPowerOfTwo(int x, int mod) { return x & (mod - 1); }//對 2 的非負整數次冪取模 31 int getBit(int a, int b) { return (a >> b) & 1; }// 獲取 a 的第 b 位,最低位編號為 0 32 int Max(int a, int b) { return b & ((a - b) >> 31) | a & (~(a - b) >> 31); }// 如果 a>=b,(a-b)>>31 為 0,否則為 -1 33 int Min(int a, int b) { return a & ((a - b) >> 31) | b & (~(a - b) >> 31); } 34 ll gcd(ll a, ll b) {return b ? gcd(b, a % b) : a;} 35 ll lcm(ll a, ll b) {return a / gcd(a, b) * b;} 36 int Abs(int n) { 37 return (n ^ (n >> 31)) - (n >> 31); 38 /* n>>31 取得 n 的符號,若 n 為正數,n>>31 等於 0,若 n 為負數,n>>31 等於 -1 39 若 n 為正數 n^0=n, 數不變,若 n 為負數有 n^(-1) 40 需要計算 n 和 -1 的補碼,然後進行異或運算, 41 結果 n 變號並且為 n 的絕對值減 1,再減去 -1 就是絕對值 */ 42 } 43 ll binpow(ll a, ll b,ll c) { 44 ll res = 1; 45 while (b > 0) { 46 if (b & 1) res = res * a%c; 47 a = a * a%c; 48 b >>= 1; 49 } 50 return res%c; 51 } 52 void extend_gcd(ll a,ll b,ll &x,ll &y) 53 { 54 if(b==0) { 55 x=1,y=0; 56 return; 57 } 58 extend_gcd(b,a%b,x,y); 59 ll tmp=x; 60 x=y; 61 y=tmp-(a/b)*y; 62 } 63 ll mod_inverse(ll a,ll m) 64 { 65 ll x,y; 66 extend_gcd(a,m,x,y); 67 return (m+x%m)%m; 68 } 69 ll eulor(ll x) 70 { 71 ll cnt=x; 72 ll ma=sqrt(x); 73 for(int i=2;i<=ma;i++) 74 { 75 if(x%i==0) cnt=cnt/i*(i-1); 76 while(x%i==0) x/=i; 77 } 78 if(x>1) cnt=cnt/x*(x-1); 79 return cnt; 80 } 81 int n,a[maxn]; 82 vector<int>pos[maxn]; 83 int cnt=0; 84 bool isprime[maxn]; 85 int prime[maxn]; 86 void judge() 87 { 88 cnt=0; 89 mem(isprime,1); 90 isprime[1]=0; 91 for(int i=2;i<maxn;i++) 92 { 93 if(isprime[i]) 94 prime[cnt++]=i; 95 for(int j=0;j<cnt&&(i*prime[j])<maxn;j++) 96 { 97 isprime[i*prime[j]]=0; 98 if(i%prime[j]==0) break; 99 } 100 } 101 } 102 void dec(int p) 103 { 104 int n=a[p]; 105 for(int i=0;i<cnt&&prime[i]*prime[i]<=n;i++) 106 { 107 if(n%prime[i]==0) 108 { 109 pos[prime[i]].push_back(p); 110 while(n%prime[i]==0) n/=prime[i]; 111 } 112 } 113 if(n>1) pos[n].push_back(p); 114 } 115 void dec_(int p) 116 { 117 int n=a[p]; 118 for(int i=2;i*i<=n;i++) 119 { 120 if(n%i==0) 121 { 122 pos[i].push_back(p); 123 while(n%i==0) n/=i; 124 } 125 } 126 if(n>1) pos[n].push_back(p); 127 } 128 int main() 129 { 130 judge(); 131 for(int i=0;i<cnt;i++) pos[prime[i]].push_back(0); 132 scanf("%d",&n); 133 for(int i=1;i<=n;i++) 134 { 135 scanf("%d",&a[i]); 136 dec(i); 137 } 138 ll ans=0; 139 for(int i=0;i<cnt;i++) 140 { 141 for(std::size_t k=1;k<pos[prime[i]].size();k++) 142 { 143 ans+=(ll)(n-pos[prime[i]][k]+1)*(pos[prime[i]][k]-pos[prime[i]][k-1]); 144 } 145 } 146 printf("%lld\n",ans); 147 return 0; 148 }
View Code