คำชี้แจงปัญหา
กำหนดอาร์เรย์ของ 0 และ 1 งานคือการค้นหาจำนวน swap ขั้นต่ำที่จำเป็นสำหรับการจัดกลุ่มทั้งหมด 1 รายการในอาร์เรย์เข้าด้วยกัน
ตัวอย่าง
หากอาร์เรย์อินพุต ={1, 0, 1, 1, 0, 1} จำเป็นต้องมีการสลับ 1 ครั้ง เช่น สลับ 0 ตัวแรกกับ 1 ตัวสุดท้าย
อัลกอริทึม
- นับจำนวนรวมของ 1 ในอาร์เรย์
- หากนับเป็น x เราก็ต้องหาอาร์เรย์ย่อยของความยาว x ของอาร์เรย์นี้ด้วยจำนวนสูงสุดที่ 1
- Swap ขั้นต่ำที่ต้องการจะเป็นจำนวน 0 ใน subarray ของความยาว x ที่มีจำนวนสูงสุดคือ 1's
ตัวอย่าง
#include <bits/stdc++.h>
using namespace std;
int getMinSwaps(int *arr, int n) {
int oneCnt = 0;
for (int i = 0; i < n; ++i) {
if (arr[i] == 1) {
++oneCnt;
}
}
int x = oneCnt;
int maxOnes = INT_MIN;
int preCompute[n] = {0};
if (arr[0] == 1) {
preCompute[0] = 1;
}
for (int i = 1; i < n; ++i) {
if (arr[i] == 1) {
preCompute[i] = preCompute[i - 1] + 1;
} else {
preCompute[i] = preCompute[i - 1];
}
}
for (int i = x - 1; i < n; ++i) {
if (i == (x - 1)) {
oneCnt = preCompute[i];
} else {
oneCnt = preCompute[i] - preCompute[i - x];
} if (maxOnes < oneCnt) {
maxOnes = oneCnt;
}
}
int swapCnt = x - oneCnt;
return swapCnt;
}
int main() {
int arr[] = {1, 0, 1, 1, 0, 1};
int n = sizeof(arr) / sizeof(arr[0]);
cout << "Minimum swap count = " << getMinSwaps(arr, n) << endl;
return 0;
} เมื่อคุณคอมไพล์และรันโปรแกรมข้างต้น มันสร้างผลลัพธ์ต่อไปนี้ -
ผลลัพธ์
Minimum swap count = 1