网易2019校招编程题--丰收

it2022-05-05  107

又到了丰收的季节,恰逢小易去牛牛的果园里游玩。 牛牛常说他对整个果园的每个地方都了如指掌,小易不太相信,所以他想考考牛牛。 在果园里有N堆苹果,每堆苹果的数量为ai,小易希望知道从左往右数第x个苹果是属于哪一堆的。 牛牛觉得这个问题太简单,所以希望你来替他回答。

输入描述:

第一行一个数n(1 <= n <= 105)。第二行n个数ai(1 <= ai <= 1000),表示从左往右数第i堆有多少苹果第三行一个数m(1 <= m <= 105),表示有m次询问。第四行m个数qi,表示小易希望知道第qi个苹果属于哪一堆。

输出描述:

m行,第i行输出第qi个苹果属于哪一堆。

示例1

输入

5 2 7 3 4 9 3 1 25 11

输出

1 5 3算法:前缀和,二分 #include<iostream> #include<unordered_map> #include<algorithm> #include<vector> using namespace std; int n,m; int main(void){ cin>>n; vector<int>f(n),s(n); for(int i=0;i<n;i++)cin>>f[i]; s[0]=f[0]; for(int i=1;i<n;i++)s[i]=s[i-1]+f[i]; cin>>m; vector<int>q(m); for(int i=0;i<m;i++)cin>>q[i]; for(int i=0;i<m;i++){ int l=0,r=n-1; while(l<r){ int mid=l+r>>1; if(s[mid]>=q[i])r=mid; else l=mid+1; } cout<<l+1<<endl; } return 0; }

 

转载于:https://www.cnblogs.com/programyang/p/11147614.html

相关资源:DirectX修复工具V4.0增强版

最新回复(0)