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

贝茜经营了一家餐厅。她的餐厅一共有 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 <bits/stdc++.h>
using namespace std;
#define int long long
#define ios ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
typedef pair<int, int> 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<<n<<endl;
            k=0;
            continue;
        }
        int l=1,r=n;
        while (l<r)
        {
            int mid=(l+r)>>1;
            if (s[mid]>=x) r=mid;
            else l=mid+1;
        }
        if (s[l]==x) cout<<n-l<<endl;
        else cout<<n-l+1<<endl;
        k=x;
    }
    return 0;
}

  • 7
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值