HDU 5726 线段树+dp

题意:给出一个序列,后q次询问,求给定区间gcd及整个序列有多少个序列的gcd和这个值相同

首先线段树维护区间gcd建树,之后预处理出每个gcd有多少个子序列,这时需要dp,

dp[i][tmp]表示以第i个数a[i]结尾的序列中有dp[i][tmp]个连续子序列的gcd值为tmp,dp[i]是一个map

那么,dp[i][tmp]如何求?显然,tmp是由dp[i-1]中所有gcd值与a[i]求gcd得到的,(因为是连续子序列的gcd值,所以由a[i-1]转移而来)。

这里加个滚动数组优化的思想,只需要两个unordered_map即可,最后用一个map的ans记录每个gcd值有多少个序列即可,复杂度应该位于o(nlogn)到o(n(logn)^2)之间

#include <bits/stdc++.h>
using namespace std;
using ll=long long;
const int maxn=1e5+5;
ll gcd[maxn<<2];
ll a[maxn];
void build(int l,int r,int rt)
{
    if(l==r){
        gcd[rt]=a[l];
        return;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    gcd[rt]=__gcd(gcd[rt<<1],gcd[rt<<1|1]);
}
ll query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)return gcd[rt];
    int m=(l+r)>>1;
    ll res=0;
    if(L<=m)res=query(L,R,l,m,rt<<1);
    if(R>m)res=__gcd(res,query(L,R,m+1,r,rt<<1|1));
    return res;
}
unordered_map<ll,ll>ans;
unordered_map<ll,ll>dp[2];
int main()
{
    int t,ca=1;
    cin>>t;
    while (t--)
    {
        int n;
        cin>>n;
        for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
        dp[0].clear();
        dp[1].clear();
        build(1,n,1);
        dp[0][a[1]]=1;
        ans[a[1]]=1;
        for(int i=2;i<=n;i++)
        {
            ans[a[i]]++;
            dp[(i&1)^1][a[i]]=1;
            for(auto xy=dp[i&1].begin();xy!=dp[i&1].end();xy++)
            {
                ll tmp=__gcd(xy->first,a[i]);
                dp[(i&1)^1][tmp]+=xy->second;
                ans[tmp]+=xy->second;
            }
            dp[i&1].clear();
        }
        int q;
        cin>>q;
        printf("Case #%d:\n",ca++);
        while (q--)
        {
            int l,r;
            scanf("%d%d",&l,&r);
            ll res=query(l,r,1,n,1);
            printf("%lld %lld\n",res,ans[res]);
        }
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/xusirui/p/9649181.html
今日推荐