codeforces 481 div3 C. Letters

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/A7_RIPPER/article/details/81737467

这里写图片描述这里写图片描述

题目大意:

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;
}

猜你喜欢

转载自blog.csdn.net/A7_RIPPER/article/details/81737467