chore(CPlusPlus): added new array program (#479)
parent
2cc1d5ffde
commit
928baa858f
|
@ -0,0 +1,64 @@
|
||||||
|
/*
|
||||||
|
Description: Index of first 1 in a sorted array of 0s and 1s.
|
||||||
|
Given a sorted array consisting 0’s and 1’s. The task is to find the index of first ‘1’ in the given array.
|
||||||
|
|
||||||
|
Approach: Use binary search to solve in least time complexity (O log n).
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <iostream>
|
||||||
|
#include <algorithm>
|
||||||
|
using namespace std;
|
||||||
|
int firstIndex(int a[], int n) {
|
||||||
|
|
||||||
|
int low=0;
|
||||||
|
int high=n-1;
|
||||||
|
|
||||||
|
while(low<=high) {
|
||||||
|
int mid = (low+high)/2;
|
||||||
|
//if mid appears to be 0, update low as mid+1 index because we want to search 1 in array
|
||||||
|
if(a[mid]==0){
|
||||||
|
low=mid+1;
|
||||||
|
}
|
||||||
|
//checking if mid element is 1 and the element at mid-1 is 0
|
||||||
|
//if it is so, then we have found index of first 1
|
||||||
|
//simply return mid
|
||||||
|
else if(a[mid]==1 && (mid==0 || a[mid-1]==0)){
|
||||||
|
return mid;
|
||||||
|
}
|
||||||
|
//updating the high to mid-1
|
||||||
|
//because the element at mid-1 index is not zero
|
||||||
|
else if (a[mid]==1){
|
||||||
|
high=mid-1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//if index not found, return -1
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
int main() {
|
||||||
|
int n;
|
||||||
|
cout << "Enter number of elements in an array\n";
|
||||||
|
cin>>n;
|
||||||
|
int a[n];
|
||||||
|
cout<<"Enter array elements: (only 0 and 1 in sorted order)\n";
|
||||||
|
for(int i=0;i<n;i++){
|
||||||
|
cin>>a[i];
|
||||||
|
}
|
||||||
|
//if someone forgets to pass sorted array
|
||||||
|
sort(a,a+n);
|
||||||
|
int ind = firstIndex(a,n);
|
||||||
|
cout<<"Index of first 1 in sorted array: "<<ind;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
Sample Input:
|
||||||
|
Enter number of elements in an array
|
||||||
|
4
|
||||||
|
Enter array elements: (only 0 and 1 in sorted order)
|
||||||
|
0
|
||||||
|
0
|
||||||
|
0
|
||||||
|
1
|
||||||
|
|
||||||
|
Output:
|
||||||
|
Index of first 1 in sorted array: 3
|
||||||
|
*/
|
|
@ -16,6 +16,7 @@
|
||||||
12. [Leaders In The Array](Arrays/leaders-in-array.cpp)
|
12. [Leaders In The Array](Arrays/leaders-in-array.cpp)
|
||||||
13. [Elements appear thrice In The Array](Arrays/Elements_appears_thrice.cpp)
|
13. [Elements appear thrice In The Array](Arrays/Elements_appears_thrice.cpp)
|
||||||
14. [Maximum Difference](Arrays/maximum-difference.cpp)
|
14. [Maximum Difference](Arrays/maximum-difference.cpp)
|
||||||
|
15. [Occurrence of one in sorted array](algorithms/CPlusPlus/Arrays/occurence-of-one-in-sorted-array.cpp)
|
||||||
|
|
||||||
## Dynamic-Programming
|
## Dynamic-Programming
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue