From c225c03e129949d1381543c6aa066fb7ae303a40 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=BB=84=E6=B5=B7?= <10402852@qq.com>
Date: Fri, 8 Mar 2024 11:25:07 +0800
Subject: [PATCH] 'commit'

---
 .../ZhongGaoJi/LanQiao15STEMA202401/5.cpp     | 59 +++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 TangDou/LanQiaoBei/ZhongGaoJi/LanQiao15STEMA202401/5.cpp

diff --git a/TangDou/LanQiaoBei/ZhongGaoJi/LanQiao15STEMA202401/5.cpp b/TangDou/LanQiaoBei/ZhongGaoJi/LanQiao15STEMA202401/5.cpp
new file mode 100644
index 0000000..d0c342f
--- /dev/null
+++ b/TangDou/LanQiaoBei/ZhongGaoJi/LanQiao15STEMA202401/5.cpp
@@ -0,0 +1,59 @@
+#include <bits/stdc++.h>
+using namespace std;
+const int N = 10010;
+int a[N];
+int sum, avg, total;
+int main() {
+    int n;
+    cin >> n;
+
+    for (int i = 1; i <= n; i++) {
+        cin >> a[i];
+        sum += a[i];
+    }
+    if (sum % n) {
+        cout << -1 << endl;
+        exit(0);
+    }
+    avg = sum / n; // 平均数
+
+    /*
+    算法思路
+    1、求出平均数avg
+    2、枚举每个数字:
+        (1) 如果当前数字比avg大，不断向后面找出比avg小的数，将自己多出的部分传递给这个数=min(超过avg的量,少于avg的量)
+        如果一下子没用了，就继续向后做同样的操作
+        (2) 如果当前数字比avg小，不断向后面找出比avg大的数，将自己少出的部分传递给这个数=min(超过avg的量,少于avg的量)
+        如果一下子没用了，就继续向后做同样的操作
+
+    举栗子思考：
+    10 1 1
+    5 1 6
+    1 5 6
+    */
+    for (int i = 1; i <= n; i++) {
+        int d = a[i] - avg; // 差值，注意：这个差值可能是正的，也可能是负的
+        total += d;
+
+        int j = i + 1;
+        while (d != 0 && j <= n) {
+            // 向后找出,比avg小的
+            if (d > 0 && a[j] < avg) {
+                int x = min(d, avg - a[j]);
+                a[i] -= x;
+                a[j] += x;
+                d -= x;
+            }
+            // 向后找出,比avg大的
+            if (d < 0 && a[j] > avg) {
+                int x = min(d, a[j] - avg);
+                a[i] += x;
+                a[j] -= x;
+                d += x;
+            }
+            j++;
+        }
+    }
+    cout << total << endl;
+    return 0;
+}
\ No newline at end of file