Codeforces 474B. Worms

Codeforces 474B. Worms

传送门：https://codeforces.com/problemset/problem/474/B

题目大意：
我们会得到n堆虫，每一堆有ai条，为它们按顺序编号，我们的任务是计算k个编号bj到底属于哪一堆。

主要想法：
首先拿到这个题目是想直接暴力，就把每个ai都加上它前面的a，针对每一个bj遍历一遍a，然后找出它在哪一个区。很遗憾，这个思路超时了。所以，我们得进行一些改进，我们先对bj记录序号并进行排序，然后与a一起遍历，这样就把复杂度降到了o（nlogn），o（mlogm）的层次。但这还不够好，我们想到这个b排序再遍历，很容易联想到优先队列，用优先会方便也会快一些，第二个方法用时93ms而优先只用了78ms。

	代码实现：（排序）

#include<bits/stdc++.h>
 
using namespace std;
int main()
{
    int n,k,x=0;
    cin>>n;
    vector<int>a(n);
    cin>>a[0];
    for(int i=1;i<n;i++)
    {
        int c;
        scanf("%d",&c);
        a[i]=a[i-1]+c;
    }
    cin>>k;
    vector<int> d(k);
    vector<pair<int,int> >b(k);
    for(int i=0;i<k;++i)
    {
        int c;
        scanf("%d",&c);
        b[i]={c,i};
    }
    sort(b.begin(),b.end());
    for(int i=0;i<k;)
    {
        if(a[x]>=b[i].first)
        {
            d[b[i].second]=x+1;
            i++;
        }
        else x++;
    }
    for(int i=0;i<k;i++)printf("%d\n",d[i]);
}

	代码实现：（优先队列）

#include<bits/stdc++.h>
 
using namespace std;
struct node
{
    int x,y;
    bool operator < (const node & a)const
    {
            return x>a.x;
    }
};
priority_queue<node>b;
int main()
{
    int n,k,x=0;
    cin>>n;
    vector<int>a(n);
    cin>>a[0];
    for(int i=1;i<n;i++)
    {
        int c;
        scanf("%d",&c);
        a[i]=a[i-1]+c;
    }
    cin>>k;
    vector<int> d(k);
    for(int i=0;i<k;++i)
    {
        int c;
        scanf("%d",&c);
        b.push({c,i});
    }
    while(b.size())
    {
        if(a[x]>=b.top().x)
        {
            d[b.top().y]=x+1;
            b.pop();
        }
        else x++;
    }
    for(int i=0;i<k;i++)printf("%d\n",d[i]);
}