牛客练习赛46 E 华华和奕奕学物理 (树状数组)

https://ac.nowcoder.com/acm/contest/894/E

一开始写了一个简单的模拟 通过率只有5%......

看题解真的理解了好久!!肥宅大哭orz

题解如下

最后一句:“维护6个树状数组即可”.....喵喵喵??

先学一下树状数组吧:

链接在这
https://blog.csdn.net/Small_Orange_glory/article/details/81290634

结合代码讲比较容易理解

#include<bits/stdc++.h>
using namespace std;
const int maxn=4e6+100;
const int T=3e5+10;
const int mod=1e9+7;
const int g=10;
#define ll long long
ll bit[8][maxn];
void add(ll b[],int i,ll C)  //更新树状数组
{
    while(i<maxn)
    {
        b[i]=(b[i]+C)%mod;  
        i+=i&-i;
    }
}
ll sum(ll b[],int i)   //求和 前i个元素的和
{
    ll  ans=0;
    while(i>0)
    {
        ans+=b[i];
        i-=i&-i;
    }
    return ans%mod;
}
int main()
{
    int Q;scanf("%d",&Q);
    while(Q--)
    {
        int op,v,t,m;scanf("%d%d%d",&op,&v,&t);
        if(op==1)
        {
            int V=v+g*(T-t);
            scanf("%d",&m);
            add(bit[1],V,1LL*m*v%mod*v%mod);
            add(bit[2],V,1LL*m*g*g%mod);
            add(bit[3],V,1LL*m*g*g%mod*t%mod*t%mod);
            add(bit[4],V,1LL*2*m*g*g%mod*t%mod);
            add(bit[5],V,1LL*2*m*v%mod*g%mod);
            add(bit[6],V,1LL*2*m*v%mod*g*t%mod);
        }
        else
        {
            int V=min(v+g*(T-t),maxn-1);
            ll ans=0;
            ans+=sum(bit[1],V);
            ans+=sum(bit[2],V)*t%mod*t%mod;
            ans+=sum(bit[3],V);
            ans-=sum(bit[4],V)*t%mod;
            ans+=sum(bit[5],V)*t%mod;
            ans-=sum(bit[6],V);
            ans%=mod;
            if(ans<0)
                ans+=mod;
            printf("%lld\n",ans);
        }
    }
    return 0;
}

先讲为什么维护6个数组:

因为将公式

扫描二维码关注公众号,回复: 6253498 查看本文章

拆开就会变成含有6项的多项式:

我们将符合条件的小球的分别记录到6个数组中 这样方便求和

那么什么是符合条件的小球呢?

仔细看题解里说:

若某一时刻a球比b球速度快,则a球始终比b球速度快。

所以如果末尾时间a球比b球快 那么a球始终比b球快

所以判断某一时刻速度比v小的小球有哪些 假设在这一时刻扔下一个初速度为v的小球 只需要看哪些小球在最后的时刻速度比这个初速度为v的小球速度慢就好了

那么对于每一次op==1 我们就更新一次数组 把v+g*(T-t)之后的树状数组全部更新

对于每一次op==2 我们就把前v+g*(T-t)项求和

照着模拟几遍就理解了(而我想了一晚上)

猜你喜欢

转载自www.cnblogs.com/dyhaohaoxuexi/p/10884399.html
今日推荐