（堆）CF12BCorrect Solution?

一、算法分析

没有用排序，而是用了一个小根堆，个人感觉蛮直观的。基本思路是先求题目要求的“最小数”，然后和输入的第二行相比较。求“最小数”的基本步骤分三步：

把第一行序列中最小的非0数字放到目标序列最前面
如果有第一行序列中有0，则把0跟在目标序列后面
再把剩下的数放到小根堆里面，然后从小根堆里面取数，依次放到目标序列后面

二、代码及注释

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<queue>
using namespace std;
const int N=15;
priority_queue<int,vector<int>,greater<int> > heap;     //建一个小根堆
vector<int> A,B,C;
int st[N];                                              //表示A中的那些数已用过
int get_min(){
    int min=10;
    int pos=-1;                                         //表示最小值的位置
    for(int i=0;i<A.size();i++) if(A[i] && A[i]<min) min=A[i];
    for(int i=0;i<A.size();i++) if(A[i]==min) pos=i;
    if(pos==-1){
        st[0]=1;
        return 0;
    }
    else{
        st[pos]=1;
        return min;
    }
}
void push_zero(){
    for(int i=0;i<A.size();i++) if(A[i]==0 && !st[i]) st[i]=1,C.push_back(0);
}
void push_min(){
    for(int i=0;i<A.size();i++) if(!st[i]) heap.push(A[i]);
    while(!heap.empty()){
        int t=heap.top();
        heap.pop();
        C.push_back(t);
    }
}
bool cmp(){
    if(B.size()!=C.size()) return false;
    for(int i=0;i<B.size();i++){
        if(B[i]!=C[i]) return false;
    }
    return true;
}
int main(){
    
    string a,b;
    cin>>a>>b;
    for(int i=0;i<a.size();i++) A.push_back(a[i]-'0');
    for(int i=0;i<b.size();i++) B.push_back(b[i]-'0');
    
    if(B.size()>1 && B[0]==0) {cout<<"WRONG_ANSWER";return 0;}
    
    C.push_back(get_min());                              //把非0的最小值放在前面
    push_zero();                                         //然后把0跟在后面
    push_min();                                          //然后把剩下的最小值放进去
    
    if(cmp()) cout<<"OK";
    else cout<<"WRONG_ANSWER";
    //test//for(int i=0;i<C.size();i++) cout<<C[i];
    
    
    
    return 0;
    
}