题目
Problem B: 大学宿舍
Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 61 Solved: 11
Description
在一所大学中有n幢公寓,从1标记到n,第i幢公寓有ai个房间。每个房间都有一个门牌号,从1标记到a1+a2+a3+...+an,1号表示第1幢楼第一个房间,a1号表示第1幢楼最后一个房间,a1+1号表示第2幢楼第一个房间,以此类推,a1+a2+a3+...+an表示第n幢楼最后一个房间。现在有m封信送到学校,每个信封上都有着一个门牌号,你的任务是将这些信分类,辨别它们属于第几幢楼的第几个房间
Input
第一行输入两个整数n,m(1<=m,n<=2*10^5)分别代表公寓个数和信封个数
第二行输入n个整数a1,a2..an表示每幢公寓的房间数
第三行输入m个整数b1,b2...bm表示每封信上的门牌号 ai<=10^10
Output
输出m行,每行有两个整数f,k分别表示第f幢楼的第k个房间
Sample Input
3 6
10 15 12
1 9 12 23 26 37
Sample Output
1 1
1 9
2 2
2 13
3 1
3 12
思路
这道题在处理学生宿舍房间个数的时候要用到前缀和,在对每个信查找房间的时候要对比信的门牌号和每个公寓的前缀和,因为前缀和一定是单调递增的,所以应当使用二分法。
代码
赛后发布题解:
#include <stdio.h>
#include <algorithm>
#define ll long long
using namespace std;
ll a[200010],sum[200010],b;
bool check(int mid)
{
return sum[mid]<b;
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
sum[0]=0;
for(int i=1;i<=n;i++)scanf("%lld",&a[i]),sum[i]=a[i]+sum[i-1];
while(m--)
{
scanf("%lld",&b);
int l=1,r=n,ans=0;
while(l<=r)//二分几栋
{
int mid=(l+r)/2;
if(check(mid))ans=mid,l=mid+1;
else r=mid-1;
}
//printf("%d %d\n",ans,sum[ans]);
ll ans2=b-sum[ans];//剩余
printf("%d %lld\n",ans+1,ans2);
}
}