Mã C ++ để có khoảng cách ngắn nhất từ ​​các trạm hình tròn

Giả sử chúng ta có hai số s và t, và một mảng D khác với n phần tử. Tuyến đường tròn của tàu điện ngầm Dreamland có n ga khác nhau. Chúng ta biết khoảng cách giữa tất cả các cặp trạm lân cận:D [i] là khoảng cách giữa trạm thứ i và i + 1, và D [n-1] là khoảng cách giữa (n-1) và trạm thứ 0. Chúng ta phải tìm khoảng cách ngắn nhất từ ​​s đến t.

Vì vậy, nếu đầu vào giống như s =1; t =3; D =[2, 3, 4, 9], thì đầu ra sẽ là 5.

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 -

n := size of D
Define an array arr of size (n + 1), and fill with 0
for initialize i := 1, when i <= n, update (increase i by 1), do:
   arr[i] := D[i - 1]
   sum1 := sum1 + arr[i]
if s > t, then:
   swap s and t
for initialize i := s, when i < t, update (increase i by 1), do:
   sum2 := sum2 + arr[i]
return minimum of sum2 and (sum1 - sum2)

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;
int solve(int s, int t, vector<int> D){
   int n = D.size(), sum1 = 0, sum2 = 0;
   vector<int> arr(n + 1, 0);
   for (int i = 1; i <= n; i++){
      arr[i] = D[i - 1];
      sum1 += arr[i];
   }
   if (s > t)
      swap(s, t);
   for (int i = s; i < t; i++)
      sum2 += arr[i];
   return min(sum2, sum1 - sum2);
}
int main(){
   int s = 1;
   int t = 3;
   vector<int> D = { 2, 3, 4, 9 };
   cout << solve(s, t, D) << endl;
}

Đầu vào

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

Đầu ra

5