fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. // your code goes here
  6. int n;
  7. cin>>n;
  8. vector<int>arr(n,0);
  9. for(int i=0;i<n;i++){
  10. cin>>arr[i];
  11. }
  12. int k1;
  13. cin>>k1;
  14. int p;
  15. for(int i=0;i<n;i++){
  16. if(arr[i]==k1){
  17. p=i;
  18. break;
  19. }
  20. }
  21. int a;
  22. for(int i=p-1;i>=0;i--){
  23. if(arr[i]>p){
  24. a=i;
  25. break;
  26. }
  27. }
  28. int b;
  29. for(int i=p+1;i<n;i++){
  30. if(arr[i]>p){
  31. b=i;
  32. break;
  33. }
  34. }
  35. int x=0;
  36. int y=0;
  37. int c=p-1;
  38. int d=p+1;
  39. for(int i=a;i<=c;i++){
  40. x++;
  41. }
  42. for(int j=d;j<=b;j++){
  43. y++;
  44. }
  45. int count=0;
  46. count=(x+1)*(y+1);
  47. cout<<"The number of subarrays with max element k is:"<<count;
  48. return 0;
  49. }
Success #stdin #stdout 0s 5320KB
stdin
5
8 2 5 1 10
5
stdout
The number of subarrays with max element k is:9