1191: 华哥倒酒

https://www.oj.swust.edu.cn/problem/show/1191

Time Limit: 5000 MS Memory Limit: 131072 KB

Submit Status Discuss

Description

    今天华哥请吃饭,待会儿打完比赛不要跑了。

    华哥现在已经去饭店了,他在桌子上整齐地摆放了n个空酒杯,编号为i酒杯的容量为ri。他有m条理(jie)由(kou)要往某写杯子里加酒。第j条理由是往编号为sj到编号为tj的杯子中倒入dj的酒。但是华哥并被注意到,在他祭出他的某条理由后,有些杯子中的酒就会漫出来!(已经倒入的酒的量大于了杯子的容量)然而,猥(ji)琐(zhi)的上决╇ф早已经看穿了这一切。他想知道,华哥的哪条理由会造成第一次漫出。但是他实在是太懒了,这个问题就丢给你了……

Input

第一行包含两个正整数n,m,表示酒杯的数量和华哥的理由条数。 第二行包含n个正整数,其中第i个数为ri,表示编号为i的酒杯的容量。 接下来有m行,每行包含三个正整数dj,sj,tj,表示第j条理由,往编号为sj到编号为tj的杯子中倒入dj的酒。每行相邻的两个数之间均用一个空格隔开。酒杯编号和理由编号都从1开始。

Output

如果没有漫出的情况发生,输出“0”。 否则第一排输出一个“-1”,第二排输出第一次造成漫出的理由的编号。

4 3 2 5 4 3 2 1 3 3 2 4 4 2 4

-1 2

Hint

对于100%的数据,有1≤n,m≤4*10^5,0≤ri,dj≤10^9,1≤sj≤tj≤n。 描述第一句话反正我是不信。

 

 

起初暴力做一直超时,后来查资料了解了一些前缀和的知识。

前缀和

 

其实可以把它理解为数学上的数列的前n项和(对于一个一维数组的前缀和)。

我们定义对于一个数组a的前缀和数组s,s[i] = a[1]+a[2]+...+a[i].

 

二维前缀和

与一维前缀和类似,设s[i][j]表示所有a[i'][j']的和。(1≤i'≤i,1≤j'≤j)

有一点像“矩形的面积”那样,把一整块区域的值都加起来。

 

前缀和的用途

一般用来求区间和。

对于一维情况,现在我给出一个数列a,要求你回答m次询问,每次询问下标j到k的和。朴素的做法显然是对于每次询问都执行一次相加操作,然后输出结果。这样做是正确的,但是当m过大时就会导致计算次数过多而有可能超时。

超时的原因一目了然,重复计算。那么我们应该怎么改进这个方法呢?想象一下,我们如果先提前算好了每一个位置的前缀和,然后用s[k]-s[j],结果不就是我们这次询问的答案吗?这样便会使计算量大大减小。

对于二维的区间和,也是类似的。


//1≤n,m≤4*10^5,线段树区间更新是可以做的

//但是这道题是简化版,若:1<=n,m<=10^6

//线段树做就可能会超时(也许会卡过,姿势要好)

//所以,可以二分答案O(logm * n)

 

#include <iostream>

#include <stdio.h>

#include <math.h>

#include <algorithm>

#include <queue>

#include <stack>

#include <vector>

#include <string>

#include <string.h>

#include <map>

#include <set>

using namespace std;

#define maxn 400005

#define inff 0x7FFFFFFF

typedef long long LL;

int a[maxn],lx[maxn],rx[maxn],val[maxn];

LL sum[maxn];//前缀和

int n,m,l,r;

int fun(int mid)

{

     for(int i=0;i<=n;i++)

        sum[i]=0;

    for(int i=1;i<=mid;i++)

    {

        sum[rx[i]]+=val[i];

        sum[lx[i]-1]-=val[i];

    }

    int re=0;

    for(int i=n;i>=1;i--)

    {

        re+=sum[i];

        if(re>a[i])

            return 1;//此区间找到溢出

    }

    return 0;//没找到

}

int main()

{

    int i;

    while(scanf("%d%d",&n,&m)!=EOF)

    {

        for(i=1;i<=n;i++)

        {

            scanf("%d",&a[i]);

            sum[i]=0;

        }

 

        for(i=1;i<=m;i++)

        {

            scanf("%d%d%d",&val[i],&lx[i],&rx[i]);

 

        }

        l=1,r=m;//对m次询问进行二分

        int flag=0;

        while(l<r)

        {

            int mid=(l+r)/2;

//            cout<<l<<" "<<r<<endl;

            if(fun(mid)==0)l=mid+1;//没找到溢出就往后区间找

            else

            {

                flag=1;

 

                r=mid;//找到溢出 往前找

            }

        }

        if(flag==1)

        {

            printf("%d\n%d\n",-1,r);

        }

        else printf("0\n");

 

    }

    return 0;

}