【排序,前缀和】Day 3 提高组模拟C组 T1 音乐节拍

题目链接

https://www.luogu.org/problemnew/show/P2969

题目大意

给定 n 种音乐,第 i 个音乐会在 B i 1 B i 1 这个播放,特别地 B 0 = 0 ,有 m 次询问,询问第 A i 个单位时间里在播放什么音乐

解题思路

排序思路

O ( n l o g n )
对询问的 A i 进行排序,预处理出所有的答案

前缀和思路

摘自 X J Q
O ( q l o g n )
对于 B i 进行前缀和处理,然#后询问时二分查找其位置,再输出

排序代码

#include<cstdio>
#include<algorithm>
using namespace std;int n,m,a[50001],j=1,c[50001];
struct node{int id,w,ans;}b[50001];
bool cmp(node x,node y){return x.w<y.w;}
int main()
{
    freopen("mnotes.in","r",stdin);
    freopen("mnotes.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",a+i),a[i]+=a[i-1];//求出每个音乐在什么时候结束
    for(int i=1;i<=m;i++) scanf("%d",&b[i].w),b[i].id=i;//读入,保存序号
    sort(b+1,b+1+m,cmp);//排序
    for(int i=1;i<=m;)
     if(b[i].w<a[j]) b[i].ans=j,i++;//如果它在第j个音乐结束之前,那么其就是在播放第j个音乐,这里其实是省略了大于a[j-1],因为这里满足单调性所以就忽略了
     else j++;//否则我们往前找到一个比它大的
    for(int i=1;i<=m;i++) c[b[i].id]=b[i].ans;//保存答案
    for(int i=1;i<=m;i++) printf("%d\n",c[i]);//输出
}

前缀和思路

#include <cstdio>
#include <cctype>
using namespace std;
int n,q,a[50005];
int in(){
    int ans=0; char c=getchar();
    while (!isdigit(c)) c=getchar();
    while (isdigit(c)) ans=ans*10+c-48,c=getchar();
    return ans;
}
int main(){
    freopen("mnotes.in","r",stdin);
    freopen("mnotes.out","w",stdout);
    n=in(); q=in(); int x;
    for (int i=1;i<=n;i++) x=in(),a[i+1]=a[i]+x;//前缀和
    while (q--){
        x=in();
        int l=1,r=n+1;
        while (l<=r){
            int mid=(l+r)>>1;
            if (x>=a[mid]) l=mid+1; else r=mid-1;//二分查找
        }
        printf("%d\n",r);
    }
    return 0; 
}

猜你喜欢

转载自blog.csdn.net/xuxiayang/article/details/80959884