codeforces 962D Merge Equals

题目链接：http://codeforces.com/contest/962/problem/D

题意：输入n，接下来输入n个数ai。每次从左往右，如果某个数字出现次数>=2，记出现次数>=2中最小的数为x，把前面那个x消除，后面的x变成2x。然后一直执行这个操作，直到无法执行，输出最后保留下来的数字，顺序和操作中的不变。

分析：一看到每次消掉最小的x，然后把下标大的x变成2x，就想要利用优先队列，id小的在队头，对于每一个数字的优先队列p[x]存储所有值为x数的id，然后优先队列q[x] pop两次，代表合并两个x，接着在2x那个队列中q[2x] push刚才两个x中的大id，直到q[x].size()<2。然后判断q[2x]的size()，继续这个操作。直到最后这些数字q[x].size()都小于2。由于数字x较大，我们需要对离散化，用map存储每一个数字离散化之后的序号，用数组b存map后的序号对应的x，即mp[x]=s，b[s]=x，x对应的优先队列为q[mp[x]]。当把数字消除更新之后，每个队列中最多只有一个id，然后每个队列q[i]对应的数字就是b[i]。按照id从小到大，输出size()=1的数字就可以了，这个比较简单。

AC代码：

#include<bits/stdc++.h>

using namespace std;

priority_queue<int,vector<int>,greater<int> >q[1000005];
map<long long ,int>mp;
long long a[1000005];
long long b[1000005];
int id[1000005],dd[1000005];
bool cmp(int x,int y){
    return id[x]<id[y];
}
int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n;
    memset(id,0x3f3f3f3f,sizeof(id));
    cin>>n;
    int ans=0;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        if(mp[a[i]]==0){
            ans++;
            mp[a[i]]=ans;
            b[ans]=a[i];
        }
        q[mp[a[i]]].push(i);
    }
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++){
        long long s=a[i];
        for(int j=1;j<=64;j++){
            if(q[mp[s]].size()<2) break;
            long long s2=s*2;
            if(mp[s2]==0){
                ans++;
                mp[s2]=ans;
                b[ans]=s2;
            }
            int d1=mp[s],d2=mp[s2];
            while(q[d1].size()>=2){
                q[d1].pop();
                int d=q[d1].top();
                q[d1].pop();
                q[d2].push(d);
            }
            s=s*2;
        }
    }
    int num=0;
    for(int i=1;i<=ans;i++){
        dd[i]=i;
        if(q[i].size()==1){
            num++;
            id[i]=q[i].top();
        }
    }
    sort(dd+1,dd+1+ans,cmp);
    cout<<num<<endl;
    for(int i=1;i<=ans;i++){
        if(id[dd[i]]==0x3f3f3f3f){
            break;
        }
        cout<<b[dd[i]]<<" ";
    }
    cout<<endl;
return 0;
}