UVALive 4097 Yungom （思路--set||优先队列的模拟）

大体题意：

给你有m 种字符，并告诉你每个字符的权值，你要组成n 个单词，n 个单词中，任意两个单词的其中一个 不是另一个单词的前缀，一个单词的权值是各个字符的权值之和！ 求组成n 个单词的最小权值？

思路：

其实这个题  稍微的类似于哈夫曼树，也不完全一样，有些许相似之处！

我们先定一个根节点 ， 根节点没有字符，  连接着m 个字符，每个叶子结点都是一个完整的单词，如果个数不足n 个单词，就在最小权值的单词后继续扩展叶子结点，扩展m 个字符，这样又会得到新的单词权值，这样不断的更新权值答案，如果发现权值变大了，那么往后肯定都没有解了，因为只会越来越大了！

这样的话，用优先队列或者set 模拟都可以！

坑：

后台数据是有0这个数据的，也就意味着有可能扩展结点  权值保持不变，因此退出循环的条件是 sum >= ans 否则一直死循环而超时！！！！

set版：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <set>
using namespace std;
const int inf = 0x3f3f3f3f;
int a[207];
multiset<int>s;
multiset<int>::iterator it;
int main(){
    int n,m;
    while(~scanf("%d %d",&n, &m) && (n || m)){
        s.clear();
        for (int i = 0; i < m; ++i){
            scanf("%d", a+i);
            s.insert(a[i]);
        }
        int ans = inf;
        while(1){
            while(s.size() < n){
                int u = *(s.begin()); s.erase(s.begin());
                for (int i = 0; i < m; ++i) s.insert(u + a[i]);
            }
            int sum = 0;
            int cnt;
            for (it = s.begin(),cnt = 0; cnt < n && it != s.end(); ++cnt, ++it){
                sum += *it;
            }
//            printf("sum = %d\n",sum);
            if (sum >= ans) break; //注意是>=  超时了n次！！！
            ans = sum;
            int u = *(s.begin());
            s.erase(s.begin());
            for (int i = 0; i < m; ++i) s.insert(u + a[i]);
        }
        printf("%d\n",ans);
    }

    return 0;
}

优先队列版：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int inf = 0x3f3f3f3f;
int a[1007];
priority_queue<int,vector<int>,greater<int> >q;
priority_queue<int,vector<int>,greater<int> >tq;
int main(){
    int n,m;
    while(~scanf("%d %d",&n, &m) && (n || m)){
        while(!q.empty())q.pop();
        while(!tq.empty())tq.pop();
        for (int i = 0; i < m; ++i) {
            scanf("%d",a+i);
            q.push(a[i]);
        }
        if (n == 1){
            printf("%d\n",q.top());
            continue;
        }
        if (m == 1 && n > 1) {
            printf("0\n");
            continue;

        }
        int ans = inf;
        while(1){
            while(q.size() < n){
                int u = q.top(); q.pop();
                for (int i = 0; i < m; ++i) q.push(u + a[i]);
            }
            int sum = 0;
            for (int i = 0; i < n; ++i){
                int u = q.top();
                tq.push(u); q.pop();
                sum += u;
            }
            if (sum >= ans) break;
            ans = sum;
            int u = tq.top();
            tq.pop();
            for (int i = 0; i < m; ++i) q.push(u + a[i]);
            while(!tq.empty()) {
                q.push(tq.top());
                tq.pop();
            }


        }
        printf("%d\n",ans);

    }

    return 0;
}