【NOIP2015模拟11.5】JZOJ8月5日提高组T2 Lucas的数列

【NOIP2015模拟11.5】JZOJ8月5日提高组T2 Lucas的数列

题目

PS：$n\ast n\ast T\ast T<=10^{18}$而不是$10^1\ast 8$

题解

题意：

给出$n$个元素的复杂度和位置
然后每次询问一个区间
每次询问给出$x,y,z$
求$x$到$y$中复杂度小于等于$z$的元素的位置按照一定计算过程后的答案（具体见题面）

分析：

看到题目给出的式子十分的繁琐
我们来化简一下
$$K=(\sum _{i=1}^m(x_i-p{)}^2)\ast m=(\sum _{i=1}^m((x_i{)}^2-2x_{i}p+p^2))\ast m=(\sum _{i=1}^m(x_i{)}^2-\sum _{i=1}^{m}2x_{i}p+\sum _{i=1}^m{p}^2)\ast m$$
$$=(\sum _{i=1}^m(x_i{)}^2-\sum _{i=1}^{m}2x_i\frac{{\sum }_{i=1}^m{x}_i}{m}+m(\frac{{\sum }_{i=1}^m{x}_i}{m}{)}^2)\ast m=m\sum _{i=1}^m(x_i{)}^2-2(\sum _{i=1}^m{x}_i{)}^2+(\sum _{i=1}^m{x}_i{)}^2=m\sum _{i=1}^m(x_i{)}^2-(\sum _{i=1}^m{x}_i{)}^2$$
所以说，$K$其实一直都是个整数：$m{\sum }_{i=1}^m(x_i{)}^2-({\sum }_{i=1}^m{x}_i{)}^2$
再看，这题并不要求在线
所以可以离线
按照$w$和$z$为第一关键字排序
然后维护一个$j$使得$1$~$j$内的$w$都小于当前的$z$
由于$z$是单调递增的，所以$j$不用清零
然后构造一棵线段树（树状数组）就可以了

Code

#include
#include
using namespace std;
struct node1
{
    long long p,w,id;
}a[400005];
struct node2
{
    long long x,y,z,lixian;
}c[400005];
struct node3
{
    long long sum1,sum2,num;
}tree[1600005];
bool cmp1(node1 x,node1 y)
{
    return x.w<y.w;
}
bool cmp2(node2 x,node2 y)
{
    return x.z<y.z;
}
long long n,m,i,j;
long long s1,s2,s3,ans[400005];
void build(long long now,long long l,long long r,long long pos,long long val)
{
    if (l==r)
    {
        if (l==pos)
        {
            tree[now].num=1;
            tree[now].sum1=val*val;
            tree[now].sum2=val;
        }
        return;
    }
    if (l>pos||r<pos) return;
    long long mid=(l+r)>>1;
    build(now<<1,l,mid,pos,val);
    build(now<<1|1,mid+1,r,pos,val);
    tree[now].num=tree[now<<1].num+tree[now<<1|1].num;
    tree[now].sum1=tree[now<<1].sum1+tree[now<<1|1].sum1;
    tree[now].sum2=tree[now<<1].sum2+tree[now<<1|1].sum2;
}
void query(long long now,long long l,long long r,long long p,long long q)
{
    if (tree[now].num==0) return;
    if (l>q||r<p) return;
    if (l>=p&&r<=q)
    {
        s1+=tree[now].num;
        s2+=tree[now].sum1;
        s3+=tree[now].sum2;
        return;
    }
    long long mid=(l+r)>>1;
    query(now<<1,l,mid,p,q);
    query(now<<1|1,mid+1,r,p,q);
}
int main()
{
    freopen("sequence.in","r",stdin);
    freopen("sequence.out","w",stdout);
    scanf("%lld%lld",&n,&m);
    for (i=1;i<=n;i++)
    {
        scanf("%lld%lld",&a[i].w,&a[i].p);
        a[i].id=i;
    }
    for (i=1;i<=m;i++)
    {
        scanf("%lld%lld%lld",&c[i].x,&c[i].y,&c[i].z);
        c[i].lixian=i;
    }
    sort(a+1,a+n+1,cmp1);
    sort(c+1,c+m+1,cmp2);
    j=1;
    for (i=1;i<=m;i++)
    {
        while (a[j].w<=c[i].z&&j<=n)
        {
            build(1,1,n,a[j].id,a[j].p);
            j++;
        }
        s1=s2=s3=0;
        query(1,1,n,c[i].x,c[i].y);
        if (s1==0) ans[c[i].lixian]=-1;
        else ans[c[i].lixian]=s1*s2-s3*s3;
    }
    for (i=1;i<=m;i++)
    {
        if (ans[i]==-1) printf("empty\n");
        else printf("%lld\n",ans[i]);
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}