Codeforces-962D-Merge Equals (模拟+stl)

D. Merge Equals
time limit per test:2 seconds
memory limit per test:256 megabytes
input:standard input
output:standard output

You are given an array of positive integers. While there are at least two equal elements, we will perform the following operation. We choose the smallest value x that occurs in the array 2 or more times. Take the first two occurrences of x in this array (the two leftmost occurrences). Remove the left of these two occurrences, and the right one is replaced by the sum of this two values (that is,2x).

Determine how the array will look after described operations are performed.

For example, consider the given array looks like [3,4,1,2,2,1,1]

. It will be changed in the following way: [3,4,1,2,2,1,1]  [3,4,2,2,2,1]  [3,4,4,2,1]  [3,8,2,1].

If the given array is look like [1,1,3,1,1]

it will be changed in the following way: [1,1,3,1,1]  [2,3,1,1]  [2,3,2]  [3,4].
Input

The first line contains a single integer n(2n150000) — the number of elements in the array.

The second line contains a sequence from nelements a1,a2,,an (1ai109) — the elements of the array.

Output

In the first line print an integer k— the number of elements in the array after all the performed operations. In the second line print k integers — the elements of the array after all the performed operations.

Examples
Input
7
3 4 1 2 2 1 1
Output
4
3 8 2 1 
Input
5
1 1 3 1 1
Output
2
3 4 
Input
5
10 40 20 50 30
Output
5
10 40 20 50 30 
Note

The first two examples were considered in the statement.

In the third example all integers in the given array are distinct, so it will not change.


题意自己看吧。。。

思路:对于每个数出现的位置我们可以都用map存起来,map套优先队列,优先队列存位置。然后从mp的第一个开始扫,对于每一个数,按题意操作直到这个数的个数<=1,新产生的数要加到map里面去,如果这个数的个数变成了0,就把map里面的这个数erase,最后map里面剩的数就是最后的结果,按下标大小从小到大输出就行。

AC代码:

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const LL maxn=150000+10;
LL arr[maxn],n;
map<LL,priority_queue<LL,vector<LL>,greater<LL> > > mp;
struct node
{
    LL a,b;
    node(){};
    node(LL aa,LL bb){a=aa;b=bb;}
    bool operator < (const node k)const{
        return a>k.a;
    }
};
priority_queue<node> que;
int main()
{
    scanf("%lld",&n);
    for(LL i=1;i<=n;i++){
        scanf("%lld",&arr[i]);
        mp[arr[i]].push(i);
    }
    map<LL,priority_queue<LL,vector<LL>,greater<LL> > >::iterator it,tmp;
    for(it=mp.begin();it!=mp.end();){
        while(it->second.size()>1){
            it->second.pop();
            LL tmp=it->second.top();
            mp[(it->first)*2].push(tmp);
            it->second.pop();
        }
        tmp=it;it++;
        if(tmp->second.size()==0) mp.erase(tmp);
    }
    printf("%d\n",mp.size());
    for(it=mp.begin();it!=mp.end();it++){
        que.push(node(it->second.top(),it->first));
    }
    while(que.size()){
        printf("%lld",que.top().b);
        que.pop();
        if(que.size()) printf(" ");
        else printf("\n");
    }
    return 0;
}


猜你喜欢

转载自blog.csdn.net/qq_37171272/article/details/79903533