AcWing 刷题计划：0x10 基本数据结构—— AcWing 149. 荷马史诗

1.题目描述：


2.题意：
略。

3.思路：
贪心。【K叉哈夫曼树】 看哈夫曼树的定义就知道是哈夫曼编码，而所谓k进制就是k叉哈夫曼树的意思。那么如何去构建一个哈夫曼树呢，先队列放入所有的数，选取最小的k个数，合并成一个节点，然后放回队列，直到剩下只有一个，这样最少使用次数的就会在树的靠下端，这样可以使得带权路径长度最短。 不过要考虑一点问题，就是对于最靠近根的那一层，如果不满足k个结点，那么我把下面层任意一个提到靠近根的那层可以降低带权路径长度，所以我们加入空结点，凑成刚好满的k叉树。由于我们询问的是：带权路径长度，以及深度。所以我们记录下深度和带权路径长度。

4.代码：

//AcWing 149. 荷马史诗
//#include
//#pragma GCC optimize(3,"Ofast","inline")
#include
#include
#include
#include
#include
#include
#include
//#include
#include
#include
#include
#include
#include
#include
#define FAST ios::sync_with_stdio(false)
#define DEV_RND ((int)rand()*RAND_MAX+rand())
#define RND(L,R) (DEV_RND%((R)-(L)+1)+(L))
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i
#define per(i,n,a) for(int i=n-1;i>=a;--i)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define li inline
#define re register
using namespace std;
//typedef uniform_int_distribution RNDI;
typedef pair<int,int> PII;
typedef vector<int> VI;
typedef double db;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
const int Hash = 131;//13331
const int maxn = 1e5+5;
const int maxm = 100000+5;
const int mod = 1e9+7;
const int inf = 0x3f3f3f3f;
const double eps = 1e-7;
const double pi = acos(-1);
//int dir[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
//li int f(int x){return x==par[x]?par[x]:par[x]=f(par[x]);}
//mt19937 eng(time(0));
//li int RND(int L,int R){RNDI rnd(L,R);return rnd(eng);}
li ll lowbit(ll x){return x&(-x);}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){if(!b){d=a,x=1,y=0;}else{ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
li ll qpow(ll a,ll b,ll MOD=mod){ll res=1;a%=MOD;while(b>0){if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res%MOD;}
li ll qmul(ll a,ll b,ll MOD=mod){return (a*b-(ll)((long double)a/MOD*b)*MOD+MOD)%MOD;}
li ll inv(ll x,ll p){return qpow(x,p-2,p);}
li ll jos(ll n,ll k,ll s=1){ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
db f(db x){return x;}
li db sim(db l,db r){return (f(l)+4.*f((l+r)/2.)+f(r))*(r-l)/6.;}
db asr(db l,db r,db ans,db eps){db m=l+(r-l)/2.,L=sim(l,m),R=sim(m,r);return fabs(L+R-ans)<=15.*eps?L+R+(L+R-ans)/15.:asr(l,m,L,eps/2)+asr(m,r,R,eps/2);}
db asr(db l,db r,db eps){return asr(l,r,sim(l,r),eps);}
namespace IO
{li ll read(){ll x=0,sign=1;char c=getchar();while(c>'9'||c<'0') {if(c=='-') sign=-1;c=getchar();}while('0'<=c&&c<='9') x=x*10+c-'0',c=getchar();return x*sign;}template<typename T>li void write(T x,char t='\n'){if(x<0){x=-x;putchar('-');};static int sta[25];int top=0;do{sta[top++]=x%10;}while(x/=10);while(top) putchar(sta[--top]+'0');putchar(t);}
}
using namespace IO;
/*-------------head-------------*/
//
typedef pair<ll,ll> PLL;
ll n,k;
priority_queue<PLL,vector<PLL>,greater<PLL>> q;
li void solve()
{k=read();rep(i,0,n){ll x=read();q.push(mp(x,0ll));}while((n-1)%(k-1)) q.push(mp(0,0)),n++;ll res=0,depth=0;while(sz(q)>1){ll s=0;rep(i,0,k){auto t=q.top();q.pop();s+=t.fi;depth=max(depth,t.se);}res+=s;q.push(mp(s,depth+1));}write(res);write(q.top().fi);//puts("");
}
int main()
{//srand(time(0));//freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);//for(int QwQ=read();QwQ;QwQ--) solve();while(~scanf("%lld",&n)) solve();return 0;
}

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！