split-array-by-prime-indices

java

python3

python

javascript

typescript

csharp

golang

kotlin

swift

rust

ruby

dart

scala

elixir

erlang

racket

cangjie

bash

mysql

mssql

postgresql

oraclesql

pythondata

Algorithms

You are given an integer array <code>nums</code>.

Split <code>nums</code> into two arrays <code>A</code> and <code>B</code> using the following rule:

<ul>
	<li>Elements at prime indices in <code>nums</code> must go into array <code>A</code>.</li>
	<li>All other elements must go into array <code>B</code>.</li>
</ul>

Return the absolute difference between the sums of the two arrays: <code>|sum(A) - sum(B)|</code>.

Note: An empty array has a sum of 0.

&nbsp;
Example 1:

<div class="example-block">
Input: nums = [2,3,4]

Output: 1

Explanation:

<ul>
	<li>The only prime index in the array is 2, so <code>nums[2] = 4</code> is placed in array <code>A</code>.</li>
	<li>The remaining elements, <code>nums[0] = 2</code> and <code>nums[1] = 3</code> are placed in array <code>B</code>.</li>
	<li><code>sum(A) = 4</code>, <code>sum(B) = 2 + 3 = 5</code>.</li>
	<li>The absolute difference is <code>|4 - 5| = 1</code>.</li>
</ul>
</div>

Example 2:

<div class="example-block">
Input: nums = [-1,5,7,0]

Output: 3

Explanation:

<ul>
	<li>The prime indices in the array are 2 and 3, so <code>nums[2] = 7</code> and <code>nums[3] = 0</code> are placed in array <code>A</code>.</li>
	<li>The remaining elements, <code>nums[0] = -1</code> and <code>nums[1] = 5</code> are placed in array <code>B</code>.</li>
	<li><code>sum(A) = 7 + 0 = 7</code>, <code>sum(B) = -1 + 5 = 4</code>.</li>
	<li>The absolute difference is <code>|7 - 4| = 3</code>.</li>
</ul>
</div>

&nbsp;
Constraints:

<ul>
	<li><code>1 &lt;= nums.length &lt;= 105</code></li>
	<li><code>-109 &lt;= nums[i] &lt;= 109</code></li>
</ul>

第 161 场双周赛

biweekly-contest-161

splitArray

高级工程师

Split Array by Prime Indices

Array

Math

Number Theory

给你一个整数数组 <code>nums</code>。

根据以下规则将 <code>nums</code> 分割成两个数组 <code>A</code> 和 <code>B</code>：

<ul>
	<li><code>nums</code> 中位于&nbsp;质数 下标的元素必须放入数组 <code>A</code>。</li>
	<li>所有其他元素必须放入数组 <code>B</code>。</li>
</ul>

返回两个数组和的&nbsp;绝对&nbsp;差值：<code>|sum(A) - sum(B)|</code>。

质数&nbsp;是一个大于 1 的自然数，它只有两个因子，1和它本身。

注意：空数组的和为 0。

&nbsp;

示例 1:

<div class="example-block">
输入: nums = [2,3,4]

输出: 1

解释:

<ul>
	<li>数组中唯一的质数下标是 2，所以 <code>nums[2] = 4</code> 被放入数组 <code>A</code>。</li>
	<li>其余元素 <code>nums[0] = 2</code> 和 <code>nums[1] = 3</code> 被放入数组 <code>B</code>。</li>
	<li><code>sum(A) = 4</code>，<code>sum(B) = 2 + 3 = 5</code>。</li>
	<li>绝对差值是 <code>|4 - 5| = 1</code>。</li>
</ul>
</div>

示例 2:

<div class="example-block">
输入: nums = [-1,5,7,0]

输出: 3

解释:

<ul>
	<li>数组中的质数下标是 2 和 3，所以 <code>nums[2] = 7</code> 和 <code>nums[3] = 0</code> 被放入数组 <code>A</code>。</li>
	<li>其余元素 <code>nums[0] = -1</code> 和 <code>nums[1] = 5</code> 被放入数组 <code>B</code>。</li>
	<li><code>sum(A) = 7 + 0 = 7</code>，<code>sum(B) = -1 + 5 = 4</code>。</li>
	<li>绝对差值是 <code>|7 - 4| = 3</code>。</li>
</ul>
</div>

&nbsp;

提示:

<ul>
	<li><code>1 &lt;= nums.length &lt;= 105</code></li>
	<li><code>-109 &lt;= nums[i] &lt;= 109</code></li>
</ul>