解法1(DP解)
以下の説明では,x の約数の個数を σ(x) と表現します.
以下の S,L を考えます.
Si,g : Aiが最右である連続部分列で,最大公約数が g となるようなものの部分列の長さの合計
Li,g : Aiが最右である連続部分列で,最大公約数が g となるようなものの個数
Si,g は g が Ai の約数でない場合は明らかに 0 であるため, 1≤i≤N∑g∈Aiの約数∑g×Si,g が求める答えです.よって,Si,g が求まっていれば,O(N×σ(Ai)) で答えを求めることができます.
次に,S,L を求める方法を考えます.
S,L について,以下の式が成り立ちます.
Si,g=gcd(Ai,g′)=g∑(Si−1,g′+Li−1,g′)
(ただし g=Ai の場合はさらに +1)
Li,g=gcd(Ai,g′)=g∑Li−1,g′
(ただし g=Ai の場合はさらに +1)
従って,
Si,gcd(Ai,g′) += Si−1,g′ + Li−1,g′
Li,gcd(Ai,g′) += Li−1,g′
と更新することで Si,Li が求まります.Si−1,g′,Li−1,g′ は g′ が Ai−1 の約数でない場合は明らかに 0 であるため,計算量は O(σ(Ai−1)logAi) です.従って,S,L は O(N×σ(Ai−1)logAi) で求まります.
ここまで,ある i について, Si,g のうち 0 でないものを緩く O(σ(Ai−1)) 個と見積もってきましたが,実はもう少し小さいオーダーで上から抑えることができます.Ai は,O(logAi) 個の素因数を持ちます.部分列の長さを増やしていったとき,その区間の最大公約数の素因数の種類数,素因数の数はともに単調減少します.よって, Si,g のうち 0 でないものは O(logAi) 個です.
以上より,O(Nlog2Ai) で答えを求めることができます.
(解法は違いますが,類題として,Ai の制約が 1≤Ai≤105 と緩い代わりに,数列ではなく木上のパスで同じものを求めるabc248_gもこの機会に解いておきましょう.)
解法2(セグ木を使った解法)
testerのあぷりしあです。D問題と同じように、セグ木上の二分探索を用いて解く解法を紹介します。
以下、g(l,r)=gcd(Al,Al+1,…Ar) と定義します。
l を固定して考えましょう。このとき、各 r に対して、g(l,r) は広義単調減少します。
g(l,r) の値が減少する境目の r (つまり、g(l,r)<g(l,r+1) を満たす r )を考えると、その個数は高々 A[l] の約数の個数となります。
そこで A[l] の各約数について、セグ木上の二分探索によって r を求め、r についてまとめて計算することでかなり計算量を改善することができます。(Dの解説も参考にしてください。)
しかしながら、この計算量は O(N×d(A)logNlogA) (A は数列A の任意の要素、d(A) はAの約数の個数)であり、d(A) が最悪で 1344 個もあるため間に合いません。
そこで、さらなる工夫として、境目が r に対する g(l,r) をすべての約数について調べるのではなく、都度更新していくことを考えます。
最初、x=A[l] として、以下を境目が配列の端に到達するまで繰り返します。
- x についての r を求め、ans に計算値を加える。
- xを、gcd(l,r+1) に更新する。
更新回数は、解法1と同様に g(l,r) の素因数の個数に着目することで O(logA) であることがわかります。
以上より、O(NlogNlog2A) で答えを求めることができます。
解法1のコード例(Python)
解法2のコード例(C++)