Codeforces Round #481 (Div. 3) ---- C. Letters (前缀和)

这里写图片描述

题意: 给你n个宿舍,其中每个宿舍都有ai个房间(1<=i<=n) ,房间从第一个宿舍起,编号从1到a1+a2+……+an. 然后给你m封信,bj(1<=j<=m) 表示给编号bj的房间的信。现在让你输出每封信属于哪个宿舍,以及所在那个宿舍的房间编号。

思路:
使每封信都有一个前缀编号,这个前缀编号是它前面的一个房间的编号,这样就方便O(1)就可以求每封信了。
复杂度:O(nlogn)
AC代码:

#include<bits/stdc++.h>
using namespace std;
#define rep(i,s,e)      for(int i=s;i<=e;i++)
#define rev(i,s,e)      for(int i=e;i>=s;i--)
#define all(x)          x.begin(),x.end()
#define sz(x)           x.size()
#define szz(x)          int(x.size()-1)
const int INF = 0x3f3f3f3f;
const int MOD = 1e9+7;
const int MAXN = 2e5+10;
typedef long long LL;
LL d[MAXN];
LL r[MAXN];
LL sum[MAXN];
int main()
{
    #ifdef LOCAL
    freopen("in.txt","r",stdin);
    #endif // LOCAL
    ios_base::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    int n,m;
    cin>>n>>m;
    d[0] = 0;
    rep(i,1,n) cin>>d[i],d[i] = d[i]+d[i-1];
    rep(i,0,m-1) cin>>r[i];
    sum[0] = 0;
    rep(i,0,m-1)
    {
        int pos = lower_bound(d,d+n+1,r[i])-d-1;//要找当前room的前缀,所以-1
        sum[i] = pos;
    }
    rep(i,0,m-1)
    {
        cout<<sum[i]+1<<" "<<r[i]-d[sum[i]]<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/m0_37624640/article/details/80302825