【LOJ6620】「THUPC 2019」不等式 / inequality(线段树)

点此看题面

大致题意: 给你两个长度为\(n\)的数组\(a_i\)\(b_i\),定义\(f_k(x)=\sum_{i=1}^k|a_ix+b_i|\),对于\(k=1\sim n\)的每个\(f_k\),求\(f_k\)的最小值。

前置知识

先考虑一个简单的问题:

\(Problem\ 1\)

给你一个长度为\(n\)的数组\(b_i\),求\(\sum_{i=1}^n|x+b_i|\)的最小值。

显然根据初中数学可知,当\(x\)\(b_i\)中位数的相反数时,这个式子取最小值。

再考虑一个升级版的问题:

\(Problem\ 2\)

给你两个长度为\(n\)的数组\(a_i\)\(b_i\),求\(\sum_{i=1}^n|a_ix+b_i|\)的最小值。(假定任意\(a_i>0\)

这看似复杂,实际上不难发现,\(|a_ix+b_i|=a_i|x+\frac{a_i}{b_i}|\),也就是一个\(|a_ix+b_i|\)可以拆成\(a_i\)\(|x+\frac{a_i}{b_i}|\)

然后按上面\(Problem\ 1\)的方法做即可。

虽说上面我们假定\(a_i>0\),但实际上对于\(a_i<0\)的情况,我们可以同时将\(a_i\)\(b_i\)变为其相反数,这样就能满足\(a_i>0\)了。

还有,题目没有特殊说明不存在\(a_i=0\)的情况,但数据中确实不存在这样的情况,我的做法遇上这种情况可能要加一些特殊处理,然而我懒得写了

前置处理

我们可以发现,这道题要我们做的,其实就是每次加入一组\(a_i\)\(b_i\),维护\(Problem\ 2\)的答案。

假如我们把\(|a_ix+b_i|\)拆成\(a_i\)\(|x+\frac{a_i}{b_i}|\),由于\(a_i<10^5\),显然最后数的个数的规模是难以接受的。

但如果我们能够开一个数组\(p\),以\(\frac{a_i}{b_i}\)为下标,那么我们实际上每次只要把\(p_{\frac{a_i}{b_i}}\)加上\(a_i\)就能很方便地维护数组。

可是,以一个实数为下标显然是不可能的。因此,我们就需要在处理询问前先将\(\frac{a_i}{b_i}\)给离散化。

注意,为了防止挂精度,推荐在排序比较分数大小时可以先十字相乘再比较,避免出现除法。

线段树

设我们求出中位数的值是\(e\),且其在离散化后的值是\(k\),那么答案就是:

\[(e\times\sum_{i=1}^kp_i-\sum_{i=1}^k p_i\times Fact_i)+(\sum_{i=k}^n p_i\times Fact_i-p_k\times\sum_{i=k}^np_i)\]

其中\(Fact_i\)表示\(i\)所表示的真实值。

则我们需要一个数据结构,能够实现四种操作:单点修改、求中位数、求区间\(\sum p_i\)和求区间\(\sum p_i\times Fact_i\)

于是我们就想到线段树。

注意求区间\(\sum p_i\times Fact_i\)看似棘手,但由于我们是单点修改,因此我们只要对于每个位置维护两个值即可。

特殊地,根据先前定义我们可知,每次加入一组\(a_i,b_i\),设\(t\)\(\frac {b_i}{a_i}\)离散化后的值,那么我们会将\(p_{t}\)加上\(a_i\),那么\(p_t\times Fact_t\)其实就是加上了\(a_i\times Fact_t=a_i\times \frac{b_i}{a_i}=b_i\)

而对于求中位数,其实我们如果维护\(g=\sum p_i\),那么中位数就是第\(\lfloor\frac{g+1}2\rfloor\)个数。

然后,我们在线段树上二分,便可以找到中位数离散化后的值\(k\),进而就可以得到\(e\)了。

至于具体实现,可见代码。

代码

#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 500000
#define LL long long
#define DB double
#define abs(x) ((x)<0?-(x):(x))
using namespace std;
int n,a[N+5],b[N+5],s[N+5],p[N+5];
I bool cmp(CI x,CI y) {return 1LL*b[x]*a[y]<1LL*b[y]*a[x];}
class FastIO
{
    private:
        #define FS 100000
        #define tc() (A==B&&(B=(A=FI)+fread(FI,1,FS,stdin),A==B)?EOF:*A++)
        #define tn (x<<3)+(x<<1)
        #define D isdigit(c=tc())
        int f;char c,*A,*B,FI[FS];
    public:
        I FastIO() {A=B=FI;}
        Tp I void read(Ty& x) {x=0,f=1;W(!D) f=c^'-'?1:-1;W(x=tn+(c&15),D);x*=f;}
}F;
class SegmentTree//线段树
{
    private:
        #define PT CI l=1,CI r=n,CI rt=1
        #define LT l,mid,rt<<1
        #define RT mid+1,r,rt<<1|1
        #define PU(x) (T[x]=T[x<<1]+T[x<<1|1],S[x]=S[x<<1]+S[x<<1|1])
        LL T[N<<2],S[N<<2];
    public:
        I void Upt(CI x,Con LL& y,Con LL& z,PT)//单点修改
        {
            if(l==r) return (void)(T[rt]+=y,S[rt]+=z);int mid=l+r>>1;
            x<=mid?Upt(x,y,z,LT):Upt(x,y,z,RT),PU(rt);
        }
        I int Qmid(Con LL& rk,PT)//求中位数
        {
            if(l==r) return l;int mid=l+r>>1;
            return rk<=T[rt<<1]?Qmid(rk,LT):Qmid(rk-T[rt<<1],RT);
        }
        I LL Qtot(CI x,CI y,PT)//区间求和1
        {
            if(x<=l&&r<=y) return T[rt];int mid=l+r>>1;
            return (x<=mid?Qtot(x,y,LT):0)+(y>mid?Qtot(x,y,RT):0);
        }
        I LL Qsum(CI x,CI y,PT)//区间求和2
        {
            if(x<=l&&r<=y) return S[rt];int mid=l+r>>1;
            return (x<=mid?Qsum(x,y,LT):0)+(y>mid?Qsum(x,y,RT):0);
        }
}S;
int main()
{
    RI i,k;LL g=0;DB e,w;for(F.read(n),i=1;i<=n;++i) F.read(a[i]);for(i=1;i<=n;++i) F.read(b[i]);//读入
    for(i=1;i<=n;++i) a[i]<0&&(a[i]=-a[i],b[i]=-b[i]),s[i]=i;sort(s+1,s+n+1,cmp);//方便起见,将a[i]取正
    for(k=0,i=1;i<=n;++i) (!k||(1LL*b[s[k]]*a[s[i]])^(1LL*b[s[i]]*a[s[k]]))&&(s[++k]=s[i]),p[s[i]]=k;//离散化,排序后去重
    for(i=1;i<=n;++i)
    {
        S.Upt(p[i],a[i],b[i]),k=S.Qmid((g+=a[i])+1>>1),e=1.0*b[s[k]]/a[s[k]];//线段树上查询中位数
        w=(e*S.Qtot(1,k)-S.Qsum(1,k))+(S.Qsum(k,n)-e*S.Qtot(k,n)),printf("%.8lf\n",w);//利用线段树查询所需信息计算答案
    }return 0;
}

猜你喜欢

转载自www.cnblogs.com/chenxiaoran666/p/LOJ6620.html