这道题我最开始数组开小了......
废话少说,进入正题
题目描述
WHM摆了N堆石子,他有点累,不想合并这些石子,所以他问了ZYC一个非常简单的问题:从左往右数第k个石子在哪一堆里?现在ZYC把这个问题分享给大家一起开心开心。
输入
输入第一行是一个数N,表示石子堆数。
第二行N个用空格隔开的数ai,表示从左往右第i堆有多少个石子。
第三行是一个数M,表示有M次询问。
第四行M个用空格隔开的数qi,表示WHM希望知道第qi个石子属于哪一堆。
输出
输出共有M行,每一行表示第qi个石子属于哪一堆。
样例输入
5 2 7 3 4 9 3 1 25 11
样例输出
1 5 3
提示
对于100%的数据,1<=N<=10^6 1<= M <=10^5 1<=ai<=1000。
题解:一个二分,简单至极,(如果我的话伤害到了你,恳请原谅......),直接给代码吧,但是,注意必须用二分!!!否则时间超限!!!
代码如下:
#include<cstdio> #include<iostream> using namespace std; int a[1000005]; int sum[1000005]; int ask[100005]; int main() { //freopen("stone.in","r",stdin); //freopen("stone.out","w",stdout); int n; cin>>n; for(int i=1;i<=n;i++) { scanf("%d",&a[i]); sum[i]=sum[i-1]+a[i]; } int m; cin>>m; for(int i=1;i<=m;i++) { int p; scanf("%d",&p); int l=1,r=n; while(l<r) { int mid=(l+r)/2; if(p<=sum[mid]) { r=mid; } else { l=mid+1; } } printf("%d ",l); } return 0; }