原题直通车:HDU 2527 Safe Or Unsafe
Safe Or Unsafe
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 1267 Accepted Submission(s): 496
Problem Description
Javac++ 一天在看计算机的书籍的时候,看到了一个有趣的东西!每一串字符都可以被编码成一些数字来储存信息,但是不同的编码方式得到的储存空间是不一样的!并且当储存空间大于一定的值的时候是不安全的!所以Javac++ 就想是否有一种方式是可以得到字符编码最小的空间值!显然这是可以的,因为书上有这一块内容--哈夫曼编码(Huffman Coding);一个字母的权值等于该字母在字符串中出现的频率。所以Javac++ 想让你帮忙,给你安全数值和一串字符串,并让你判断这个字符串是否是安全的?
Input
输入有多组case,首先是一个数字n表示有n组数据,然后每一组数据是有一个数值m(integer),和一串字符串没有空格只有包含小写字母组成!
Output
如果字符串的编码值小于等于给定的值则输出yes,否则输出no。
Sample Input
2 12 helloworld 66 ithinkyoucandoit
Sample Output
no yes
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
typedef pair<int,int> pii;
const int maxn=1000005;
int f[maxn], num[27], fa[maxn], g[maxn];
char s[maxn];
struct cmp {
bool operator()(const pii &t1,const pii &t2) {
return t1.first>t2.first;
}
};
int find_fa(int x) {
if(fa[x]==x) return x;
int ret=find_fa(fa[x]);
g[x]+=g[fa[x]];
return fa[x]=ret;
}
int main() {
int T; scanf("%d",&T);
while(T--) {
for(int i=0; i<maxn; ++i) fa[i]=i;
memset(num, 0, sizeof(num));
memset(g, 0, sizeof(g));
int n; scanf("%d%s",&n,s);
int len=strlen(s);
for(int i=0; i<len; ++i) num[s[i]-'a']++;
priority_queue< pii, vector<pii>, cmp >M;
int t=0, u=0;
for(int i=0; i<26; ++i)
if(num[i]) f[t]=num[i], M.push(make_pair(num[i],t++)), u++;
int ans=(u==1?f[0]:0);
while(!M.empty()) {
pii rt=M.top(); M.pop();
if(M.empty()) break;
int x=rt.first, a=rt.second;
rt=M.top(); M.pop();
int y=rt.first, b=rt.second;
fa[a]=fa[b]=t;
f[t]=a+b, M.push(make_pair(x+y,t++));
}
for(int i=0; i<t; ++i) if(fa[i]!=i) g[i]++;
for(int i=0; i<u; ++i) {
find_fa(i);
ans+=f[i]*g[i];
}
if(ans<=n) puts("yes");
else puts("no");
}
return 0;
}