【並查集分塊】Codeforces 475D CGCDSSQ

題目點這裏


bin神說這題和他當年出的一道題差不多TAT

聽了半天終於聽懂了TAT。。。

利用gcd的遞減性 從左往右把一段的gcd和最新的數相同的的放到一個並查集裏去。。

可以證明段數是不會增加的 = =


#include <cstdio>
#include <iostream>
#include <map>

using namespace std;

const int Nmax = 1e5 + 5;

int N, Q;
int a[Nmax], f[Nmax], g[Nmax];
map <int, long long> m;

int find(int x) { return (x == f[x]) ? x : f[x] = find(f[x]); }
inline void Union(int u, int v) { f[find(u)] = find(v); }
inline int gcd(int a, int b)
{
    if (!b) return a;
    for (int t = a % b; t; a = b, b = t, t = a % b);
    return b;
}

int main()
{
    ios :: sync_with_stdio(false);
    cin >> N;
    for (int i = 1; i <= N; ++ i) cin >> a[i];
    for (int i = 1; i <= N + 1; ++ i) f[i] = i;
    
    for (int i = N; i; -- i) {
        int s = i, last = i;
        g[i] = a[i];
        for (int j = find(i + 1); j <= N; j = find(j + 1)) {
            g[j] = gcd(a[i], g[j]);
            if (g[j] ^ g[last]) {
                m[g[last]] += last - s + 1;
                s = last + 1;
            }
            else Union(last, j);
            last = j;
        }
        m[g[N]] += N - s + 1;
    }
    
    for (cin >> Q; Q --; ) {
        int x; cin >> x;
        cout << m[x] << endl;
    }
    
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章