Cpp14程序通过数字左移最小次数来最大化素数和非素数数组元素之和之间的差异
给定一个大小为N的数组arr[] ,任务是通过将数组元素的数字左移最少 1 来找到数组中存在的素数之和与非素数之和之间的最大差次数。
例子:
Input: arr[] = {541, 763, 321, 716, 143}
Output: Difference = 631, Count of operations = 6
Explanation:
Operation 1: Left shift the digits of arr[1] (= 763). Therefore, arr[1] becomes 637.
Operation 2: Left shift the digits of arr[1] (= 637). Therefore, arr[1] becomes 376.
Operation 3: Left shift the digits of arr[2] (= 321). Therefore, arr[2] becomes 213.
Operation 4: Left shift the digits of arr[2] (= 213). Therefore, arr[2] becomes 132.
Operation 5: Left shift the digits of arr[3] (= 716). Therefore, arr[3] becomes 167.
Operation 6: Left shift the digits of arr[4] (= 143). Therefore, arr[4] becomes 431.
Therefore, Sum of prime array elements = 541 + 167 + 431 = 1139.
Therefore, sum of non-prime array elements = 376 + 132 = 508.
Therefore, difference = 1139 – 508 = 631.
Input: {396, 361, 359, 496, 780}
Output: Difference = 236, Count of operations = 4
Explanation:
Operation 1: Left shift the digits of arr[1] (= 361). Therefore, arr[1] becomes 613.
Operation 2: Left shift the digits of arr[2] (= 359). Therefore, arr[2] becomes 593.
Operation 3: Left shift the digits of arr[4] (= 780). Therefore, arr[4] becomes 807.
Operation 4: Left shift the digits of arr[4] (= 807). Therefore, arr[4] becomes 078.
Therefore, required difference = 613 + 593 – 496 – 78 – 396 = 236.
方法:给定的问题可以贪婪地解决。如果可以将一个元素转换为一个或多个素数,则取其中的最大值。否则,尝试通过使用所有可能的旋转来最小化元素。
请按照以下步骤解决问题:
- 初始化两个变量,比如ans和cost,分别存储所需的最大差异和最小操作数。
- 使用变量i遍历数组arr[]并执行以下步骤:
- 将变量maxPrime和minRotation初始化为-1以存储可以从arr[i]通过左旋转获得的最大质数和最小数。
- 生成数字arr[i]的所有左旋转。
- 如果arr[i]是超过maxPrime的素数,则将maxPrime更新为arr[i]和相应的成本。
- 如果maxPrime的值保持不变,则通过类似地生成所有左旋转来找到minRotation的值。
- 将arr[i]的值添加到ans 。
- 完成以上步骤后,打印ans和cost的值作为结果。
下面是上述方法的实现:
C++
// C++ program for the above approach
#include
using namespace std;
// Function to check if a
// number is prime or not
bool isPrime(int n)
{
// Base cases
if (n <= 1)
return false;
if (n <= 3)
return true;
// Check if the number is
// a multiple of 2 or 3
if (n % 2 == 0 || n % 3 == 0)
return false;
int i = 5;
// Iterate until square root of n
while (i * i <= n) {
// If n is divisible by both i and i + 2
if (n % i == 0 || n % (i + 2) == 0)
return false;
i = i + 6;
}
return true;
}
// Function to left shift a number
// to maximize its contribution
pair rotateElement(int n)
{
// Convert the number to string
string strN = to_string(n);
// Stores the maximum prime number
// that can be obtained from n
int maxPrime = -1;
// Store the required
// number of operations
int cost = 0;
string temp = strN;
// Check for all the left
// rotations of the number
for (int i = 0; i < strN.size(); i++) {
// If the number is prime, then
// take the maximum among them
if (isPrime(stoi(temp)) && stoi(temp) > maxPrime) {
maxPrime = stoi(temp);
cost = i;
}
// Left rotation
temp = temp.substr(1) + temp[0];
}
int optEle = maxPrime;
// If no prime number can be obtained
if (optEle == -1) {
optEle = INT_MAX;
temp = strN;
// Check all the left
// rotations of the number
for (int i = 0; i < strN.size(); i++) {
// Take the minimum element
if (stoi(temp) < optEle) {
optEle = stoi(temp);
cost = i;
}
// Left rotation
temp = temp.substr(1) + temp[0];
}
optEle *= (-1);
}
return { optEle, cost };
}
// Function to find the maximum sum
// obtained using the given operations
void getMaxSum(int arr[], int N)
{
// Store the maximum sum and
// the number of operations
int maxSum = 0, cost = 0;
// Traverse array elements
for (int i = 0; i < N; i++) {
int x = arr[i];
// Get the optimal element and the
// number of operations to obtain it
pair ret = rotateElement(x);
int optEle = ret.first, optCost = ret.second;
// Increment the maximum difference
// and number of operations required
maxSum += optEle;
cost += optCost;
}
// Print the result
cout << "Difference = " << maxSum << " , "
<< "Count of operations = " << cost;
}
// Driven Program
int main()
{
// Given array arr[]
int arr[] = { 541, 763, 321, 716, 143 };
// Store the size of the array
int N = sizeof(arr) / sizeof(arr[0]);
// Function call
getMaxSum(arr, N);
return 0;
}
Difference = 631 , Count of operations = 6
时间复杂度: O(N*√X*log(X)),其中X是数组中的最大元素
辅助空间: O(1)
有关更多详细信息,请参阅完整的文章通过将数字左移最小次数来最大化素数和非素数数组元素之和之间的差异!