餐厅——前缀和 + 二分查找

贝茜经营了一家餐厅。她的餐厅一共有 n 张桌子,编号 1∼n。其中,第 i 张桌子有 ai 个座位。

这一天,她的餐厅陆续来了 q 波客人。其中,第 i 波客人有 ki 个人。

当一波客人到来时,贝茜会给客人们安排座位。贝茜是个强迫症,她总是希望第 1 张桌子坐满后,再安排第 2 张桌子,第 2 张桌子坐满后,再安排第 3 张桌子,以此类推。
也就是说,在编号较小的桌子坐满之前,贝茜不会安排编号较大的桌子坐人。
一波客人有可能被安排在不止一张桌子上落座。
如果一波客人的数量大于或等于当时所有桌子的剩余空位数量,那么他们会将所有桌子都坐满,坐不下的人会自行离开。

每安排完一波到来的客人后,贝茜都会审查一遍餐厅:
如果当前餐厅尚有桌子没有被坐满,贝茜就会照常营业,准备迎接下一波客人。
如果当前餐厅所有桌子均已被坐满,贝茜就会赶走餐厅内的所有客人,清空所有桌子,并准备迎接下一波客人。

对于 1 ≤ i ≤ q,请你计算并回答,贝茜第 i 次审查完餐厅后,餐厅还有多少个桌子没有被坐满。

注意,如果贝茜在某一次审查时,清空了所有桌子,那么这次审查后,餐厅就还有 n 个桌子没有被坐满。

输入
第一行包含两个整数 n,q (1 ≤ n,q ≤ 2e5).
第二行包含 n 个整数 a1,a2,…,an(1 ≤ ai ≤ 1e9).
第三行包含 q 个整数 k1,k2,…,kq(1 ≤ ki ≤ 1e14).

输出
共 q 行,其中第 i 行输出贝茜第 i 次审查完餐厅后,餐厅中没有被坐满的桌子数量。

输入样例1:
5 5
1 2 1 2 1
3 10 1 1 1
输出样例1:
3
5
4
4
3

输入样例2:
4 4
1 2 3 4
9 1 10 6
输出样例2:
1
4
4
1

#include 
using namespace std;
#define int long long
#define ios ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
typedef pair PII;
const double PI = acos(-1.0);
const int N = 2e6 + 10;
int a[N],s[N];
int n,q;
signed main()
{
    ios;
    cin>>n>>q;
    for (int i=1;i<=n;i++) cin>>a[i];
    for (int i=1;i<=n;i++) s[i]=s[i-1]+a[i];
    int k=0;
    while (q--)
    {
        int x;
        cin>>x;
        x +=k;
        if (x>=s[n]) 
        {
            cout<>1;
            if (s[mid]>=x) r=mid;
            else l=mid+1;
        }
        if (s[l]==x) cout<

你可能感兴趣的:(算法,数据结构)