Pythonで配列の素元の合計が素数であるかどうかを確認します

配列numsがあるとします。指定された配列内のすべての素元の合計も素元であるかどうかを確認する必要があります

したがって、入力がnums =[1,2,4,5,3,3]のようである場合、すべての素数の合計が（2 + 5 + 3 + 3）=13であり、13もであるため、出力はTrueになります。プライム。

これを解決するには、次の手順に従います-

• MAX：=10000
• ふるい：=サイズMAXのリストとtrueで埋める
• 関数generate_list_of_primes（）を定義します
• sieve [0]：=False、sieve [1]：=False
• 2〜MAX-1の範囲のiの場合は、
  • sieve [i]が真の場合、
    • 2 ^ iからMAXの範囲のjの場合、iを増やします
      • sieve [j]：=False
• メインの方法から次の手順を実行します。
• generate_list_of_primes（）
• 合計：=0
• 範囲0からarr-1のサイズのiの場合、do
  • sieve [arr [i]]が真の場合、
    • 合計：=合計+ arr [i]
• sieve [total]がtrueの場合、
  • Trueを返す
• Falseを返す

理解を深めるために、次の実装を見てみましょう-

サンプルコード

MAX = 10000
sieve = [True] * MAX
 
def generate_list_of_primes() :
   sieve[0] = False
   sieve[1] = False
     
   for i in range(2, MAX) :
      if sieve[i] :
         for j in range(2**i, MAX, i) :
           sieve[j] = False
     
def solve(arr) :
   generate_list_of_primes()
   total = 0
   for i in range(len(arr)) :
      if sieve[arr[i]] :
         total += arr[i]

   if sieve[total] :
      return True
   return False

nums = [1,2,4,5,3,3]
print(solve(nums))

入力

[1,2,4,5,3,3]

出力

True