[NOIP模拟][数据结构][二分]Work

题目描述:
题目背景: SOURCE：NOIP2015-SHY-5
假设现在离 noip 还有 m 天，有 n 个人要去参加比赛。他们每个人都有一个预定的训练量 r[i] ，所以每一天他们都抓紧时间练习。但是由于条件限制，第 i 天只有 t[i] 的时间可以练习。
我们都知道，一个人在开始干活以前总要浪费一些时间做一些杂七杂八的事情。现在我们假定第 i 个人每天在训练前浪费的时间是固定的，记为 d[i] 。这段浪费掉的时间过后，选手会专心致志训练，他们会充分利用剩下的时间。然而一个可能的情况时，一个人还在无所事事的时候，某一天的训练时间已经过去了，所以他那一天什么事情都没有做。
现在请问每个人在第几天的时候可以完成自己的训练任务。当然会存在志向远大但是很懒惰的人到最后也是做不完的情况。
输入格式:
第一行两个整数 n，m ，表示人数和天数 。
接下来一行 m 个整数 t[i] 。
接下来 n 行每行两个整数 d[i]，r[i] 。
输出格式:
一行输出 n 个整数表示每个人在第几天可以完成自己的工作，如果完不成，输出 0 。
样例输入：
3 3
4 2 5
1 3
2 5
3 4
样例输出：
1 3 0
数据范围：
对 30% 的输入数据 ：1≤n,m≤1000
对 100% 的输入数据 ：1≤n,m≤ 200000；1≤t[i]≤1000000； 0≤d[i]≤1000000；1≤r[i]≤1000000
注意事项：
如果某人浪费的时间超过一天，不需减去负的时间。
题目分析:
对于每个人，如果你能在第i天前完成工作，那么你必然能在第j(j>i)前完成。所以可以二分天数。于是就变成判断是否能在第i天完成训练。前i天的完成量等于前i天中所有比此人d大的t的总和减去天数(大于d的天数)*d。于是我们可以用树状数组来维护总和与天数，如下：
1、先将训练时间t[i]和浪费时间d[i]分别按降序排序；
2、从最大的d[1]开始，将大于此时d的t[i]加入树状数组维护sum[i]，i代表天数，存的是此时已加入的t的和，同时维护天数times[i],目前一共有几天（这个天数就自然是大于d的天数）。
3、对于d[1]，二分天数，询问当前天数下i的sum和times,前i天的完成量等于sum[i]-times[i]*d[1],比较此结果与r的大小，继续二分，直到二分出正确值。
4、其余d以此类推，只需每次多加入一些t进入树状数组。
附代码:

#include<iostream>
#include<cstring>
#include<string>
#include<cstdlib>
#include<cstdio>
#include<ctime>
#include<cmath>
#include<cctype>
#include<iomanip>
#include<algorithm>
using namespace std;

const int N=2e5+10;
long long sum[N];
int n,m,times[N],po=1,ans[N];

struct node{
    int w;
    int num;
    int r;
}t[N],p[N];

int readint()
{
    char ch;int i=0,f=1;
    for(ch=getchar();(ch<'0'||ch>'9')&&ch!='-';ch=getchar());
    if(ch=='-') {ch=getchar();f=-1;}
    for(;ch>='0'&&ch<='9';ch=getchar()) i=(i<<3)+(i<<1)+ch-'0';
    return i*f;
}

bool comp(const node &a,const node &b)
{
    if(a.w!=b.w) return a.w>b.w;
    return a.num<b.num;
}

int lowbit(int i)
{
    return i&(-i);
}

void insert(int x,int w1,int w2)
{
    for(int i=x;i<=m;i+=lowbit(i))
        sum[i]+=w1,times[i]+=w2;
}

bool query(int x,int num)
{
    long long ret=0;
    for(int i=x;i>=1;i-=lowbit(i))
        ret+=sum[i],ret-=times[i]*p[num].w;
    if(ret>=p[num].r) return true;
    else return false;
}

int half(int x)
{
    int l=1,r=m,mid;
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(query(mid,x)==true) r=mid-1;
        else l=mid+1;
    }
    if(l>m) return 0;
    return l;
}

int main()
{
    //freopen("work.in","r",stdin);
    //freopen("work.out","w",stdout);

    n=readint();m=readint();
    for(int i=1;i<=m;i++)
    {
        t[i].w=readint();
        t[i].num=i;
    }
    for(int i=1;i<=n;i++)
    {
        p[i].w=readint();
        p[i].r=readint();
        p[i].num=i;
    }
    sort(t+1,t+m+1,comp);
    sort(p+1,p+n+1,comp);
    for(int i=1;i<=n;i++)
    {
        while(p[i].w<t[po].w&&po<=m)
        {
            insert(t[po].num,t[po].w,1);
            po++;
        }
        ans[p[i].num]=half(i);
    }
    for(int i=1;i<=n;i++) cout<<ans[i]<<" ";//printf("%d ",ans[i]);

    return 0;
}

附测试数据:
百度云链接: https://pan.baidu.com/s/1sldo7rJ 密码: mft3