勵志用少的代碼做高效表達。

---

思路分析

題意：給定n個數，問最少操作幾次，使數列中的數全部相等。
操作一：將數列中任意數減一。
操作二：將數列中任意數減一。任意數加一(相當于把1挪過去)

涉及到最少的題，一定就是貪心了， 當然，本題考思維多一點，如果只是單純的貪，會很麻煩。

首先求出平均數n

最開始的思路是：將>n的數的溢出值，挪給<n的數。 最后將多出來的數直接減掉。

思路可行， 但這不是最優化的解法。

想一想我們最終的目的：要求最后的值一樣，因此可以這樣簡化：對>n的數的溢出部分累加， 輸出累加和即可。 因為相加和相減是相對的，只要求出了>n要操作的次數， <n的部分自然就處理掉了（只考慮相對的情況，類似博弈論）。

優化前的代碼（耗時800ms）：

#include<bits/stdc++.h> using namespace std; int a[100010]; int main() {ios::sync_with_stdio(false);int T; cin>>T; while(T--) {long long n, sum = 0; cin>>n;for(int i = 0; i < n; i++) { cin>>a[i]; sum+=a[i]; }long long sum1=0, sum2=0;sum /= n;for(int i = 0; i < n; i++) if(a[i] > sum) sum1 += a[i]-sum;else sum2 += fabs(sum-a[i]);cout << (long long)(min(sum1,sum2)+fabs(sum1-sum2)) << endl;} return 0; }

注意：在第16行代碼里，變量進行運算后，其取值范圍會改變成int行變量的范圍，需加強轉，否則溢出。

---

優化后的代碼（耗時160ms）

#include<bits/stdc++.h> using namespace std; int a[100010]; int main() {ios::sync_with_stdio(false);int T; cin>>T; while(T--) {long long n, sum = 0; cin>>n;for(int i = 0; i < n; i++) { cin>>a[i]; sum+=a[i]; }long long sum1=0;sum /= n;for(int i = 0; i < n; i++) if(a[i] > sum) sum1 += a[i]-sum;cout << (long long)(sum1) << endl;} return 0; }

總結

以上是生活随笔為你收集整理的15行代码AC——ZOJ - 4118 Stones in the Bucket（思维题+优化方案）（第十届山东省ACM程序设计竞赛F题）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。