http://acm.hdu.edu.cn/showproblem.php?pid=6533

題意：一顆m層的滿n叉樹，從k數(shù)組中不重復的選擇一些數(shù)作為樹的邊，使得所有節(jié)點到根節(jié)點的距離之和最小

題解：對于一顆m層的滿n叉樹而言，某條邊重復計算的次數(shù)是該邊連接的子樹的結(jié)點的數(shù)量，因為是滿n叉樹，所以這個結(jié)點數(shù)量是可以O(shè)(1)求出的，求min時，只要貪心選擇小的權(quán)值為邊賦值，O(E)復雜度內(nèi)即可算出。而排序復雜度是O(NlogN),所以總復雜度是O(NlogN+E)。

C++版本一

/* *@Author: STZG *@Language: C++ */ #include <bits/stdc++.h> #include<iostream> #include<algorithm> #include<cstdlib> #include<cstring> #include<cstdio> #include<string> #include<vector> #include<bitset> #include<queue> #include<deque> #include<stack> #include<cmath> #include<list> #include<map> #include<set> //#define DEBUG #define RI register int #define endl "\n" using namespace std; typedef long long ll; typedef __int128 lll; const int N=200000+10; const int M=100000+10; const int MOD=1e9+7; const double PI = acos(-1.0); const double EXP = 1E-8; const int INF = 0x3f3f3f3f; ll t,n,m,k,p,l,r,u,v; ll ans,cnt,flag,temp,sum; ll a[N]; char str; struct node{}; ll POW(ll a,ll b,ll c){ll res=1;ll base=a%c;while(b){if(b&1)res=(res*base)%c;base=(base*base)%c;b>>=1;}return res; } int main() { #ifdef DEBUGfreopen("input.in", "r", stdin);//freopen("output.out", "w", stdout); #endif//ios::sync_with_stdio(false);//cin.tie(0);//cout.tie(0);//scanf("%d",&t);//while(t--){while(~scanf("%lld%lld%lld%lld",&k,&m,&n,&p)){for(int i=1;i<=k;i++)scanf("%lld",&a[i]);if(m==0||n==0){cout<<0<<endl;continue;}sort(a+1,a+k+1);int pos=n;int fac=m-1;ans=0;for(int i=1;i<=k;i++){if(!fac)break;ans=(ans+((a[i]%p)*(n==1?fac:(POW(n,fac,p)-1)/(n-1)))%p)%p;//cout<<(ll)(POW(n,fac,p)-1)/(n-1)<<endl;if(i==pos){//cout<<pos<<endl;fac--;pos=pos+POW(n,m-fac,p);}} cout<<ans<<endl;}//}#ifdef DEBUGprintf("Time cost : %lf s\n",(double)clock()/CLOCKS_PER_SEC); #endif//cout << "Hello world!" << endl;return 0; }

總結(jié)

以上是生活随笔為你收集整理的Build Tree的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。