存在于二叉树第 K 层的素数
给定一个数字K ,任务是打印在该级别存在的素数,因为所有素数都以二叉树的形式表示。
例子:
Input: K = 3
2
/ \
3 5
/\ / \
7 11 13 17
Output :7, 11, 13, 17
Explanation:
2
/ \
3 5
/\ / \
7 11 13 17
So primes present at level 3 : 7, 11, 13, 17
Input :K = 2
2
/ \
3 5
Output :3 5
朴素方法:朴素方法是构建素数的二叉树,然后获取特定级别 k 的元素。
它不适用于大量数据,因为它需要太多时间。
高效方法:假设有 n 个元素,任务是使用这 n 个元素构建二叉树,然后可以使用 log 2 n 级别构建它们。
因此,给定级别 k,如果所有素数都存在于一维数组中,则此处存在的元素为 2 k-1到 2 k -1。
因此,以下是算法:
- 使用埃拉托色尼筛法找到最大 MAX_SIZE 的素数。
- 计算level的left_index和right_index为left_index = 2 k-1 ,right_index = 2 k -1。
- 从素数数组的 left_index 到 right_index 输出素数。
C++
// CPP program of the approach
#include
using namespace std;
// initializing the max value
#define MAX_SIZE 1000005
// To store all prime numbers
vector primes;
// Function to generate N prime numbers using
// Sieve of Eratosthenes
void SieveOfEratosthenes(vector& primes)
{
// Create a boolean array "IsPrime[0..MAX_SIZE]" and
// initialize all entries it as true. A value in
// IsPrime[i] will finally be false if i is
// Not a IsPrime, else true.
bool IsPrime[MAX_SIZE];
memset(IsPrime, true, sizeof(IsPrime));
for (int p = 2; p * p < MAX_SIZE; p++) {
// If IsPrime[p] is not changed, then it is a prime
if (IsPrime[p] == true) {
// Update all multiples of p greater than or
// equal to the square of it
// numbers which are multiple of p and are
// less than p^2 are already been marked.
for (int i = p * p; i < MAX_SIZE; i += p)
IsPrime[i] = false;
}
}
// Store all prime numbers
for (int p = 2; p < MAX_SIZE; p++)
if (IsPrime[p])
primes.push_back(p);
}
void printLevel(int level)
{
cout << "primes at level " << level << ": ";
int left_index = pow(2, level - 1);
int right_index = pow(2, level) - 1;
for (int i = left_index; i <= right_index; i++) {
cout << primes[i - 1] << " ";
}
cout << endl;
}
// Driver Code
int main()
{
// Function call
SieveOfEratosthenes(primes);
printLevel(1);
printLevel(2);
printLevel(3);
printLevel(4);
return 0;
}
Java
// Java program of the approach
import java.util.*;
class GFG
{
// initializing the max value
static final int MAX_SIZE = 1000005;
// To store all prime numbers
static Vector primes = new Vector();
// Function to generate N prime numbers using
// Sieve of Eratosthenes
static void SieveOfEratosthenes(Vector primes)
{
// Create a boolean array "IsPrime[0..MAX_SIZE]" and
// initialize all entries it as true. A value in
// IsPrime[i] will finally be false if i is
// Not a IsPrime, else true.
boolean[] IsPrime = new boolean[MAX_SIZE];
for (int i = 0; i < MAX_SIZE; i++)
IsPrime[i] = true;
for (int p = 2; p * p < MAX_SIZE; p++)
{
// If IsPrime[p] is not changed, then it is a prime
if (IsPrime[p] == true)
{
// Update all multiples of p greater than or
// equal to the square of it
// numbers which are multiple of p and are
// less than p^2 are already been marked.
for (int i = p * p; i < MAX_SIZE; i += p)
IsPrime[i] = false;
}
}
// Store all prime numbers
for (int p = 2; p < MAX_SIZE; p++)
if (IsPrime[p])
primes.add(p);
}
static void printLevel(int level)
{
System.out.print("primes at level " + level + ": ");
int left_index = (int) Math.pow(2, level - 1);
int right_index = (int) (Math.pow(2, level) - 1);
for (int i = left_index; i <= right_index; i++)
{
System.out.print(primes.get(i - 1) + " ");
}
System.out.println();
}
// Driver Code
public static void main(String[] args)
{
// Function call
SieveOfEratosthenes(primes);
printLevel(1);
printLevel(2);
printLevel(3);
printLevel(4);
}
}
// This code is contributed by Rajput-Ji
Python3
# Python3 program of the approach
MAX_SIZE = 1000005
primes = []
# Function to generate N prime numbers using
# Sieve of Eratosthenes
def SieveOfEratosthenes():
# Create a boolean array "IsPrime[0..MAX_SIZE]" and
# initialize all entries it as True. A value in
# IsPrime[i] will finally be false if i is
# Not a IsPrime, else True.
IsPrime = [True] * MAX_SIZE
p = 2
while p * p < MAX_SIZE:
# If IsPrime[p] is not changed, then it is a prime
if (IsPrime[p] == True):
# Update all multiples of p greater than or
# equal to the square of it
# numbers which are multiple of p and are
# less than p^2 are already been marked.
for i in range(p * p, MAX_SIZE, p):
IsPrime[i] = False
p += 1
# Store all prime numbers
for p in range(2, MAX_SIZE):
if (IsPrime[p]):
primes.append(p)
def printLevel(level):
print("primes at level ", level, ":", end=" ")
left_index = pow(2, level - 1)
right_index = pow(2, level) - 1
for i in range(left_index, right_index + 1):
print(primes[i - 1], end=" ")
print()
# Driver Code
# Function call
SieveOfEratosthenes()
printLevel(1)
printLevel(2)
printLevel(3)
printLevel(4)
# This code is contributed by mohit kumar 29
C#
// C# program of the approach
using System;
using System.Collections.Generic;
class GFG
{
// initializing the max value
static readonly int MAX_SIZE = 1000005;
// To store all prime numbers
static List primes = new List();
// Function to generate N prime numbers using
// Sieve of Eratosthenes
static void SieveOfEratosthenes(List primes)
{
// Create a bool array "IsPrime[0..MAX_SIZE]" and
// initialize all entries it as true. A value in
// IsPrime[i] will finally be false if i is
// Not a IsPrime, else true.
bool[] IsPrime = new bool[MAX_SIZE];
for (int i = 0; i < MAX_SIZE; i++)
IsPrime[i] = true;
for (int p = 2; p * p < MAX_SIZE; p++)
{
// If IsPrime[p] is not changed, then it is a prime
if (IsPrime[p] == true)
{
// Update all multiples of p greater than or
// equal to the square of it
// numbers which are multiple of p and are
// less than p^2 are already been marked.
for (int i = p * p; i < MAX_SIZE; i += p)
IsPrime[i] = false;
}
}
// Store all prime numbers
for (int p = 2; p < MAX_SIZE; p++)
if (IsPrime[p])
primes.Add(p);
}
static void printLevel(int level)
{
Console.Write("primes at level " + level + ": ");
int left_index = (int) Math.Pow(2, level - 1);
int right_index = (int) (Math.Pow(2, level) - 1);
for (int i = left_index; i <= right_index; i++)
{
Console.Write(primes[i - 1] + " ");
}
Console.WriteLine();
}
// Driver Code
public static void Main(String[] args)
{
// Function call
SieveOfEratosthenes(primes);
printLevel(1);
printLevel(2);
printLevel(3);
printLevel(4);
}
}
// This code is contributed by 29AjayKumar
Javascript
输出:
primes at level 1: 2
primes at level 2: 3 5
primes at level 3: 7 11 13 17
primes at level 4: 19 23 29 31 37 41 43 47