Computer >> Máy Tính >  >> Lập trình >> C ++

Chương trình C ++ để tìm số lượng vị trí đặt bom tối thiểu để tiêu diệt tất cả quái vật

Giả sử chúng ta có hai mảng X và H. Cả hai đều có N phần tử, cũng có hai số khác D và A. Xét trong một câu chuyện, một con cáo bạc đang đánh nhau với N con quái vật. Các quái vật đang đứng trong một hàng, tọa độ của quái vật thứ i là X [i], và máu của nó là H [i]. Cáo bạc có thể sử dụng bom để tấn công quái vật. Thả bom tại vị trí x sẽ gây sát thương cho tất cả quái vật trong phạm vi từ x - D đến x + D. Máu của chúng sẽ giảm đi A. Khi tất cả quái vật chỉ còn 0 máu thì cáo thắng. Chúng tôi phải tìm ra mức tối thiểu có thể là cần thiết để giành chiến thắng.

Vì vậy, nếu đầu vào giống như D =3; A =2; X =[1, 5, 9]; H =[2, 4, 2], thì đầu ra sẽ là 2, vì quả bom đầu tiên ở tọa độ 4, các giá trị sức khỏe mới là [0, 2, 2], sau đó ở vị trí 6 để tạo ra tất cả các giá trị sức khỏe [0, 0, 0].

Các bước

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

Define a large array q
Define an array of x and h pairs
n := size of X
d := D
a := A
for initialize i := 1, when i <= n, update (increase i by 1), do:
   num[i].x := X[i - 1]
   num[i].h := H[i - 1]
sort the array num
sum := 0
for initialize i := 1, when i <= n, update (increase i by 1), do:
   q[i] := q[i] + q[i - 1]
   num[i].h := num[i].h - q[i] * a
   if num[i].h <= 0, then:
      Ignore following part, skip to the next iteration
   p := (if num[i].h mod a is same as 0, then num[i].h / a, otherwise num[i].h / a + 1)
   tmp := num[i].x + 2 * d
   sum := sum + p
   q[i] := q[i] + p
   l := i, r = n
   while l < r, do:
      mid := (l + r + 1) / 2
      if num[mid].x <= tmp, then:
         l := mid
      Otherwise
         r := mid - 1
      q[l + 1] - = p
return sum

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

#include <bits/stdc++.h>
using namespace std;

const int maxn = 2e5 + 20;
int n;
int d, a, q[maxn];
struct node{
   int x, h;
   bool operator<(const node& a) const{
      return x < a.x;
   }
} num[maxn];
int solve(int D, int A, vector<int> X, vector<int> H){
   n = X.size();
   d = D;
   a = A;
   for (int i = 1; i <= n; i++){
      num[i].x = X[i - 1];
      num[i].h = H[i - 1];
   }
   sort(num + 1, num + n + 1);
   int sum = 0;
   for (int i = 1; i <= n; i++){
      q[i] += q[i - 1];
      num[i].h -= q[i] * a;
      if (num[i].h <= 0)
         continue;
      int p = (num[i].h % a == 0 ? num[i].h / a : num[i].h / a + 1);
      int tmp = num[i].x + 2 * d;
      sum += p;
      q[i] += p;
      int l = i, r = n;
      while (l < r){
         int mid = (l + r + 1) >> 1;
         if (num[mid].x <= tmp)
            l = mid;
         else
            r = mid - 1;
      }
      q[l + 1] -= p;
   }
   return sum;
}
int main(){
   int D = 3;
   int A = 2;
   vector<int> X = { 1, 5, 9 };
   vector<int> H = { 2, 4, 2 };
   cout << solve(D, A, X, H) << endl;
}

Đầu vào

3, 2, { 1, 5, 9 }, { 2, 4, 2 }

Đầu ra

2