Count the number of primes in the prefix sum array of the given array
Last Updated :
31 May, 2022
Given an array arr[] of N integers, the task is to count the number of primes in the prefix sum array of the given array.
Examples:
Input: arr[] = {1, 4, 8, 4}
Output: 3
The prefix sum array is {1, 5, 13, 17}
and the three primes are 5, 13 and 17.
Input: arr[] = {1, 5, 2, 3, 7, 9}
Output: 1
Approach: Create the prefix sum array and then use Sieve of Eratosthenes to count the number of primes in the prefix sum array.
Below is the implementation of the above approach:
C++
#include <bits/stdc++.h>
using namespace std;
int primeCount( int arr[], int n)
{
int max_val = *max_element(arr, arr + n);
vector< bool > prime(max_val + 1, true );
prime[0] = false ;
prime[1] = false ;
for ( int p = 2; p * p <= max_val; p++) {
if (prime[p] == true ) {
for ( int i = p * 2; i <= max_val; i += p)
prime[i] = false ;
}
}
int count = 0;
for ( int i = 0; i < n; i++)
if (prime[arr[i]])
count++;
return count;
}
void getPrefixArray( int arr[], int n, int pre[])
{
pre[0] = arr[0];
for ( int i = 1; i < n; i++) {
pre[i] = pre[i - 1] + arr[i];
}
}
int main()
{
int arr[] = { 1, 4, 8, 4 };
int n = sizeof (arr) / sizeof (arr[0]);
int pre[n];
getPrefixArray(arr, n, pre);
cout << primeCount(pre, n);
return 0;
}
|
Java
import java.util.*;
class GFG
{
static int max_element( int a[])
{
int m = a[ 0 ];
for ( int i = 0 ; i < a.length; i++)
m = Math.max(a[i], m);
return m;
}
static int primeCount( int arr[], int n)
{
int max_val = max_element(arr);
boolean prime[] = new boolean [max_val + 1 ];
for ( int p = 0 ; p <= max_val; p++)
prime[p] = true ;
prime[ 0 ] = false ;
prime[ 1 ] = false ;
for ( int p = 2 ; p * p <= max_val; p++)
{
if (prime[p] == true )
{
for ( int i = p * 2 ; i <= max_val; i += p)
prime[i] = false ;
}
}
int count = 0 ;
for ( int i = 0 ; i < n; i++)
if (prime[arr[i]])
count++;
return count;
}
static int [] getPrefixArray( int arr[], int n, int pre[])
{
pre[ 0 ] = arr[ 0 ];
for ( int i = 1 ; i < n; i++)
{
pre[i] = pre[i - 1 ] + arr[i];
}
return pre;
}
public static void main(String args[])
{
int arr[] = { 1 , 4 , 8 , 4 };
int n = arr.length;
int pre[]= new int [n];
pre=getPrefixArray(arr, n, pre);
System.out.println(primeCount(pre, n));
}
}
|
Python3
def primeCount(arr, n):
max_val = max (arr)
prime = [ True ] * (max_val + 1 )
prime[ 0 ] = prime[ 1 ] = False
p = 2
while p * p < = max_val:
if prime[p] = = True :
for i in range (p * 2 , max_val + 1 , p):
prime[i] = False
p + = 1
count = 0
for i in range ( 0 , n):
if prime[arr[i]]:
count + = 1
return count
def getPrefixArray(arr, n, pre):
pre[ 0 ] = arr[ 0 ]
for i in range ( 1 , n):
pre[i] = pre[i - 1 ] + arr[i]
if __name__ = = "__main__" :
arr = [ 1 , 4 , 8 , 4 ]
n = len (arr)
pre = [ None ] * n
getPrefixArray(arr, n, pre)
print (primeCount(pre, n))
|
C#
using System;
class GFG
{
static int max_element( int [] a)
{
int m = a[0];
for ( int i = 0; i < a.Length; i++)
m = Math.Max(a[i], m);
return m;
}
static int primeCount( int [] arr, int n)
{
int max_val = max_element(arr);
bool [] prime = new bool [max_val + 1];
for ( int p = 0; p <= max_val; p++)
prime[p] = true ;
prime[0] = false ;
prime[1] = false ;
for ( int p = 2; p * p <= max_val; p++)
{
if (prime[p] == true )
{
for ( int i = p * 2; i <= max_val; i += p)
prime[i] = false ;
}
}
int count = 0;
for ( int i = 0; i < n; i++)
if (prime[arr[i]])
count++;
return count;
}
static int [] getPrefixArray( int [] arr, int n, int [] pre)
{
pre[0] = arr[0];
for ( int i = 1; i < n; i++)
{
pre[i] = pre[i - 1] + arr[i];
}
return pre;
}
public static void Main()
{
int [] arr = { 1, 4, 8, 4 };
int n = arr.Length;
int [] pre = new int [n];
pre = getPrefixArray(arr, n, pre);
Console.Write(primeCount(pre, n));
}
}
|
PHP
<?php
function primeCount( $arr , $n )
{
$max_val = max( $arr );
$prime = array_fill (0, $max_val + 1, true);
$prime [0] = false;
$prime [1] = false;
for ( $p = 2; $p * $p <= $max_val ; $p ++)
{
if ( $prime [ $p ] == true)
{
for ( $i = $p * 2; $i <= $max_val ; $i += $p )
$prime [ $i ] = false;
}
}
$count = 0;
for ( $i = 0; $i < $n ; $i ++)
if ( $prime [ $arr [ $i ]])
$count ++;
return $count ;
}
function getPrefixArray( $arr , $n , $pre )
{
$pre [0] = $arr [0];
for ( $i = 1; $i < $n ; $i ++)
{
$pre [ $i ] = $pre [ $i - 1] + $arr [ $i ];
}
return $pre ;
}
$arr = array ( 1, 4, 8, 4 );
$n = count ( $arr ) ;
$pre = array ();
$pre = getPrefixArray( $arr , $n , $pre );
echo primeCount( $pre , $n );
?>
|
Javascript
<script>
function primeCount(arr, n)
{
let max_val = Math.max(...arr);
let prime = new Array(max_val + 1).fill( true );
prime[0] = false ;
prime[1] = false ;
for (let p = 2; p * p <= max_val; p++) {
if (prime[p] == true ) {
for (let i = p * 2; i <= max_val; i += p)
prime[i] = false ;
}
}
let count = 0;
for (let i = 0; i < n; i++)
if (prime[arr[i]])
count++;
return count;
}
function getPrefixArray(arr, n, pre)
{
pre[0] = arr[0];
for (let i = 1; i < n; i++) {
pre[i] = pre[i - 1] + arr[i];
}
}
let arr = [ 1, 4, 8, 4 ];
let n = arr.length;
let pre = new Array(n);
getPrefixArray(arr, n, pre);
document.write(primeCount(pre, n));
</script>
|
Time Complexity: O(n + sqrt(max(arr))), where max(arr) is the largest element of the array arr.
Auxiliary Space: O(n + max(arr))
Like Article
Suggest improvement
Share your thoughts in the comments
Please Login to comment...