题目大意:
n个宿舍楼m次查询,第二行给出每个宿舍楼的房间数,第三行每次查询要求输出b号房间在第几个宿舍楼的第几号房间。每个宿舍楼的房间都是从1开始编号的。
解题思路:
维护一个前缀和数组,代表到当前宿舍楼共有多少房间可用(包括之前的宿舍楼)二分查找宿舍楼的位置即可。
代码如下:
#include<iostream>
#include<cstdio>
#include<fstream>
#include<set>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<iomanip>
#include<cstdlib>
#include<list>
#include<queue>
#include<stack>
#include<algorithm>
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
#define HASHP 13331;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
ll n,m;
ll dor[200010],len[200010];//宿舍楼房间数,前缀和数组
int binary_search(ll k)//二分查找
{
ll le=0,ri=n+1;
ll d,r;
while(le<ri)
{
ll mid=(le+ri)/2;
if(len[mid]>=k)
{
if(len[mid-1]<k)//如果当前位置满足条件
{
d=mid;
break;
}
else ri=mid;//缩小右边界
}
else le=mid;
}
return d;//返回宿舍楼号
}
int main()
{
std::ios::sync_with_stdio(false);
cin.tie(0);
//freopen("test.txt","r",stdin);
//freopen("output.txt","w",stdout);
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>dor[i];
len[i]+=dor[i]+len[i-1];
}
while(m--)
{
ll k;
cin>>k;
ll d=binary_search(k);
ll r=k-len[d-1];//房间号
cout<<d<<' '<<r<<endl;
}
return 0;
}