P1631 序列合并 题解
解题思路
为了高效地找到序列中所有出现次数
初始化频率表:使用哈希表记录每个元素的出现次数。
找到最小元素:在频率表中找到出现次数
合并操作:找到该元素在序列中的两个最小下标,删除第一个下标的元素,并将第二个下标的元素替换为两倍的该元素。
更新频率表:更新频率表中的元素出现次数。
重复步骤2-4,直到没有出现次数
Code
#include <bits/stdc++.h>
using namespace std;
int main() {
int n;
cin >> n;
vector<int> a(n);
unordered_map<int, int> freq;
for (int i = 0; i < n; ++i) {
cin >> a[i];
freq[a[i]]++;
}
while (true) {
int min_val = INT_MAX;
for (const auto& p : freq) {
if (p.second >= 2) {
min_val = min(min_val, p.first);
}
}
if (min_val == INT_MAX) break;
int count = 0;
for (int i = 0; i < n; ++i) {
if (a[i] == min_val) {
if (count == 0) {
a[i] = 0; // 标记删除
count++;
} else {
a[i] = 2 * min_val;
freq[2 * min_val]++;
break;
}
}
}
freq[min_val] -= 2;
if (freq[min_val] == 0) {
freq.erase(min_val);
}
}
vector<int> result;
for (int i = 0; i < n; ++i) {
if (a[i] != 0) {
result.push_back(a[i]);
}
}
cout << result.size() << endl;
for (int val : result) {
cout << val << " ";
}
cout << endl;
return 0;
}
时间复杂度:
空间复杂度:
本文作者:BadBadBad__AK
本文链接:https://www.cnblogs.com/BadBadBad/p/18644659/P1631
版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步