进阶实验4-3.5 哈夫曼编码 (30 分)

最新推荐文章于 2024-09-23 07:32:39 发布

二狗子的天下

最新推荐文章于 2024-09-23 07:32:39 发布

阅读量224

点赞数 1

分类专栏：中国大学MOOC-陈越、何钦铭-数据结构文章标签： c语言

本文链接：https://blog.csdn.net/weixin_47027777/article/details/120106903

版权

中国大学MOOC-陈越、何钦铭-数据结构专栏收录该内容

7 篇文章 0 订阅

订阅专栏

仅供参考

#include <iostream>
#include <cstdio>
#include <queue>
#include <string>
#include <map>
using namespace std;

const int maxn=10002;
int N;
priority_queue<int,vector<int>,greater<int> > q;
struct node{
    bool flag; 
    int left,right;
}Tree[maxn];
int Index;
bool flag;
map<char,int> mp;

int forlen(){
    int sum=0;
    while(q.size()!=1){
        int a=q.top();
        q.pop();
        int b=q.top();
        q.pop();
        sum=sum+a+b;
        q.push(a+b);
    }
    return sum;
}

void build(int &root,string s){
    if(root==-1){
        root=Index++;
        Tree[root].left=Tree[root].right=-1;
    }
    if(Tree[root].flag && (Tree[root].left!=-1 || Tree[root].right!=-1)) flag=true;
    if(s.size()==0){
        Tree[root].flag=true;
        if(Tree[root].left!=-1 || Tree[root].right!=-1) flag=true;
        return;
    }
    if(s[0]=='0'){
        s.erase(s.begin());
        build(Tree[root].left,s);
    }else{
        s.erase(s.begin());
        build(Tree[root].right,s);
    }
}

int main(void){
    for(int i=0;i<maxn;i++) Tree[i].flag=false;
    cin>>N;
    int w;
    char ch;
    for(int i=0;i<N;i++){
        cin>>ch>>w;
        q.push(w);
        mp[ch]=w;
    }
    int hufLen=forlen();
    int root=-1;
    int k,sum=0;
    cin>>k;
    for(int i=0;i<k;i++){
    	Index=0;
    	for(int i=0;i<maxn;i++) Tree[i].flag=false;
    	flag=false;
    	sum=0;
    	root=-1;
    	
        for(int j=0;j<N;j++){
            string s;
            cin>>ch>>s;
            sum+=(mp[ch]*s.size());
            build(root,s);
        }
        
        if(flag){
            printf("No\n");
            continue;
        }
        if(sum==hufLen) printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}