勵志用少的代碼做高效的表達
題目(提交)鏈接→UVa-10763
本題爲水題,因此側重點由解題轉向優化。
題意:判斷第一列的數字是否與第二列的數字相同(亂序)。解題方向多樣,值得探究:
1、map哈希表解法:定義map<int, int> cnt;
,其中cnt[i]
表示學校i對應的人數增減變化量,若有人把學校i作爲出發地,則cnt[i]--
;若作爲目的地,則cnt[i]++
。最終若是每個學校變化量均爲0,說明項目可行;否則不可行。特點:時間、空間複雜度皆爲O(n),都是動態開闢,基數小。爲上等解法。
2、vector+sort解法:用兩個vector容器分別存儲第一列和第二列的數據。輸入完成後排序,if(v1 == v2),則YES。特點:時間複雜度爲O(nlogn),空間複雜度爲O(n),但動態開闢,基數小。爲中等解法。
3、數組解法:開兩個大數組a[50W]
,b[50W]
,置0,分別存儲第一列和第二列的數字。如:輸入2 3,則a[2]++
,b[3]++
;最後從頭到尾遍歷,if(a[i] != b[i])
,則NO。
特點:時間、空間複雜度爲O(n),但基數大(50w),爲下等解法。
下等解法太差勁了,所以這裏只給出中等、上等解法。
vector+sort解法:
#include<bits/stdc++.h>
using namespace std;
int main() {
int n; while((cin >> n) && n) {
vector<int> v1, v2;
for(int i = 0; i < n; i++) {
int x1; cin >> x1; v1.push_back(x1);
int x2; cin >> x2; v2.push_back(x2);
}
sort(v1.begin(),v1.end()); sort(v2.begin(),v2.end());
cout << ((v1 == v2) ? "YES\n" : "NO\n");
}
return 0;
}
map哈希表映射解法:
#include<bits/stdc++.h>
using namespace std;
int n, a, b;
int main() {
while(cin >> n && n != 0) {
map<int, int> cnt; //每個學校對應的人數增減變化量
for(int i = 0; i < n; i++) {
cin >> a >> b;
cnt[a]--; cnt[b]++; //分別代表出去與近來。
}
bool isWork = true; //標記是否可行
for(auto p : cnt)
if(p.second != 0) {
isWork = false;
break;
}
cout << (isWork ? "YES\n" : "NO\n");
}
return 0;
}