codevs 1217 借教室

1217 借教室

#include<bits/stdc++.h>
using namespace std;
#define maxn 1000000
#define inf 1<<30

int n,m,tot,num[maxn],q[maxn];
int sta[maxn],d[maxn],end[maxn];

bool check(int now)
{
    tot=0;
    for(int i=1;i<=n;i++) q[i]=0;
    for(int i=1;i<=now;i++) q[sta[i]]+=d[i],q[end[i]+1]-=d[i];
    for(int i=1;i<=n;i++)
    {
        tot+=q[i];
        if(tot>num[i]) return false;
    }
    return true;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&num[i]);
    for(int i=1;i<=m;i++) scanf("%d%d%d",&d[i],&sta[i],&end[i]);
    int l=1,r=m;
    while(l<=r)
    {
        int mid=(l+r)/2;
        if(check(mid)) l=mid+1;
        else r=mid-1;
    }
    if(l>m) printf("0
");
    else printf("-1
%d
",l);
    return 0;
}

#include<bits/stdc++.h>
using namespace std;
const int maxn=1000005;
int n,m,x,y,num,i;

struct Edge{
    int l,r,w,f,mn;
}tree[maxn<<2];

char Cget;
inline void read(int &now)
{
    now=0; Cget=getchar();
    while(Cget>'9'||Cget<'0') Cget=getchar();
    while(Cget>='0'&&Cget<='9') now=now*10+Cget-'0',Cget=getchar();
}

inline int min(int a,int b) { return a<b?a:b;}

inline void tree_down(int k)
{
    tree[k<<1].f+=tree[k].f,tree[k<<1|1].f+=tree[k].f;
    tree[k<<1].mn-=tree[k].f,tree[k<<1|1].mn-=tree[k].f;
    tree[k].f=0;
}

void tree_build(int l,int r,int k)
{
    tree[k].l=l,tree[k].r=r;
    if(l==r)
    {
        read(tree[k].mn);
        return ;
    }
    int mid=(l+r)>>1;
    tree_build(l,mid,k<<1);
    tree_build(mid+1,r,k<<1|1);
    tree[k].mn=min(tree[k<<1].mn,tree[k<<1|1].mn);
}

void tree_change(int l,int r,int k,int to)
{
    if(tree[k].l>=l&&tree[k].r<=r)
    {
        if(tree[k].mn<to) 
        {
            printf("-1
%d
",i);
            exit(0);
        }
        tree[k].mn-=to,tree[k].f+=to;
        return ;
    }
    if(tree[k].f) tree_down(k);
    int mid=(tree[k].l+tree[k].r)>>1;
    if(r<=mid) tree_change(l,r,k<<1,to);
    else if(l>mid) tree_change(l,r,k<<1|1,to);
    else{
        tree_change(l,mid,k<<1,to);
        tree_change(mid+1,r,k<<1|1,to);
    }
    tree[k].mn=min(tree[k<<1].mn,tree[k<<1|1].mn);
}

int main()
{
    read(n),read(m);
    tree_build(1,n,1);
    for(i=1;i<=m;i++)
    {
        read(num),read(x),read(y);
        tree_change(x,y,1,num);
    }
    puts("0");
    return 0;
}