洛谷P2221 [HAOI2012] 高速公路

链接

　　https://www.luogu.org/problem/show?pid=2221

题解

　　挺水的线段树题
　　期望这两个字以前望着就害怕，现在觉得还蛮友好的…
　　题目让你随机抽两个点，并且每种情况都是等概率的，易知在区间[l,r]共有(r-l)条路径，方案数就是$C_{r-l}^2$，那么每种情况的概率都是$1\over C_{r-l}^2$。那就直接把所有可能情况的权值加起来，最后比上$C_{r-l}^2$就是你的答案。
　　现在问题成了，怎么求“随机选两个点，所有情况的权值和”。显然一个区间由一个右端点和一个长度唯一确定，考虑每条道路的权值被加了几次。一条道路(i,i+1)，在数以(i+1)为右端点的道路时被数了(i-l+1)次，在数以(i+2)为右端点的道路时也被数了(i-l+1)次，因此这条道路总共被数了$(i-l+1)(r-i)$次。
　　我们把道路的权值看做前面那个点的权值，这样方便研究。那么查询中的l,r在处理时先进行r-=1，然后再进行处理（式子好写）。
　　化简一下式子$(i-l+1)(r-i+1)$，就成了$-i^2+i(l+r)+(-l+r-lr+1)$，所以一次查询，就是求

$$\sum_{i=l}^r-i^2a_i+(l+r)ia_i+(-l+r-lr+1)a_i$$
　　线段树直接维护 $a_i$、 $ia_i$、 $i^2a_i$的和，然后就OK了。

代码

//线段树
#include <cstdio>
#include <algorithm>
#define maxn 500000
#define ll long long
using namespace std;
ll N, M, ndtot, s1[maxn], s2[maxn];
struct segtree
{
    ll s[3], tag, l, r;
    segtree* ch[2];
}pool[maxn], *root;
inline ll f(ll l, ll r, ll p)
{
    if(p==0)return (r-l+1);
    if(p==1)return s1[r]-s1[l-1];
    if(p==2)return s2[r]-s2[l-1];
}
void pushdown(segtree *p)
{
    ll i;
    if(!p->tag)return;
    if(p->ch[0])p->ch[0]->tag+=p->tag, p->ch[1]->tag+=p->tag;
    for(i=0;i<=2;i++)p->s[i]+=p->tag*f(p->l,p->r,i);
    p->tag=0;
}
void pushup(segtree *p)
{
    if(p->ch[0]==0)return;
    pushdown(p->ch[0]), pushdown(p->ch[1]);
    for(ll i=0;i<=2;i++)p->s[i]=p->ch[0]->s[i]+p->ch[1]->s[i];
}
void segtag(segtree *p, ll l, ll r, ll d)
{
    pushdown(p);
    ll mid=(p->l+p->r)>>1;
    if(l<=p->l and r>=p->r){p->tag+=d;return;}
    if(l<=mid)segtag(p->ch[0],l,r,d);
    if(r>mid)segtag(p->ch[1],l,r,d);
    pushup(p);
}
ll segsum(segtree *p, ll l, ll r, ll type)
{
    pushdown(p);
    ll mid=(p->l+p->r)>>1, s=0;
    if(l<=p->l and r>=p->r)return p->s[type];
    if(l<=mid)s+=segsum(p->ch[0],l,r,type);
    if(r>mid)s+=segsum(p->ch[1],l,r,type);
    return s;
}
void build(segtree *p, ll l, ll r)
{
    ll mid=(l+r)>>1;
    p->l=l, p->r=r;
    if(l==r)return;
    build(p->ch[0]=pool+ ++ndtot,l,mid);
    build(p->ch[1]=pool+ ++ndtot,mid+1,r);
}
ll gcd(ll a, ll b){return !b?a:gcd(b,a%b);}
void print(ll x, ll y)
{
    ll d=gcd(x,y);
    printf("%lld/%lld\n",x/d,y/d);
}
void init()
{
    ll i;
    for(i=1;i<=100000;i++)s1[i]=s1[i-1]+i, s2[i]=s2[i-1]+i*i;
}
void work()
{
    char type[5];
    ll l, r, v, i, ans;
    scanf("%lld%lld",&N,&M);
    build(root=pool+ ++ndtot,1,N);
    for(i=1;i<=M;i++)
    {
        scanf("%s%lld%lld",type,&l,&r);r--;
        if(*type=='Q')
        {
            ans=segsum(root,l,r,0)*(-l+r-l*r+1);
            ans+=segsum(root,l,r,1)*(l+r);
            ans-=segsum(root,l,r,2);
            print(ans,(r-l+2)*(r-l+1)/2);
        }
        else
        {
            scanf("%lld",&v);
            segtag(root,l,r,v);
        }
    }
}
int main()
{
    init();
    work();
    return 0;
}