费解的gcd

费解的gcd

https://ac.nowcoder.com/acm/contest/71419/F

浅浅来一个线段树的解法,希望能帮到大家

using namespace std;
const int N=1e6+9,mod=998244353;
int n,q;
int a[N],b[N],e[N];
int ne[N*4];
void pruck(int u)//求结点值
{
    ne[u]=__gcd(ne[u<<1],ne[u<<1|1]);
}
void build(int u,int l,int r)//建树
{
    if(l==r)
    {
        ne[u]=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(u<<1,l,mid);
    build(u<<1|1,mid+1,r);
    pruck(u);
}
void date(int u,int x,int l,int r)//更新结点数据
{
    if(l==r&&l==x)
    {
        ne[u]=0;
        return;
    }
    int mid=(l+r)>>1;
    if(mid>=x)
        date(u<<1,x,l,mid);
    else
        date(u<<1|1,x,mid+1,r);
    pruck(u);
}
void solve()
{
    cin>>n>>q;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    build(1,1,n);
    int ans=0;
    for(int i=1;i<=q;i++)
    {
        cin>>e[i];
        if(e[i]==0)
        {
            ans=(ans+ne[1])%mod;
        }
        else
        {
            date(1,e[i],1,n);
        }
    }
    cout<<ans<<'\n';
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    solve();
}
全部评论

相关推荐

1 收藏 评论
分享
牛客网
牛客企业服务