BZOJ4770 圖樣(概率期望+動態規劃)
阿新 • • 發佈:2018-11-18
很好 getc const con () stream lin 復雜度 turn
考慮求出所有MST的權值和再除以方案數,方案數顯然是2mn。
按位考慮,顯然應該讓MST裏的邊高位盡量為0。那麽根據最高位是0還是1將點集劃分成兩部分,整張圖的MST就是由兩部分各自的MST之間連一條最小邊得到的。兩部分的MST權值和可以dp得到,即設f[i][j]表示i個點權值在0~2j-1的MST權值和,枚舉最高位是0的點的數量k,由f[k][j-1]和f[i-k][j-1]轉移而來。問題只剩下求最小邊的權值和。
這個東西也不是很好求,考慮求最小邊不小於某值的方案數。同樣根據最高位是0還是1劃分點集成四個部分,轉移比較顯然,主要註意邊界,即所有邊該位都為1的情況,以及某邊沒有點的情況。盯著這個邊界調了一下午最後發現果然這裏根本就沒寫掛,而是預處理2k 時少了一部分。慘絕人寰。
復雜度O(n4m2m),雖然darkbzoj上只跑了3s,bzoj上還是根本卡不過去。
#include<iostream> #include<cstdio> #include<cmath> #include<cstdlib> #include<cstring> #include<algorithm> using namespace std; #define ll long long #define N 51 #define M 8 #define P 258280327 char getc(){charc=getchar();while ((c<‘A‘||c>‘Z‘)&&(c<‘a‘||c>‘z‘)&&(c<‘0‘||c>‘9‘)) c=getchar();return c;} int gcd(int n,int m){return m==0?n:gcd(m,n%m);} int read() { int x=0,f=1;char c=getchar(); while (c<‘0‘||c>‘9‘) {if (c==‘-‘) f=-1;c=getchar();} while (c>=‘0‘&&c<=‘9‘) x=(x<<1)+(x<<3)+(c^48),c=getchar(); return x*f; } int n,m,C[N][N],f[N][M+1],g[N][N][M],h[N][N][M][1<<M],p[N*M]; void inc(int &x,int y){x+=y;if (x>=P) x-=P;} int inv(int a) { int s=1; for (int k=P-2;k;k>>=1,a=1ll*a*a%P) if (k&1) s=1ll*s*a%P; return s; } int main() { #ifndef ONLINE_JUDGE freopen("bzoj4770.in","r",stdin); freopen("bzoj4770.out","w",stdout); const char LL[]="%I64d\n"; #else const char LL[]="%lld\n"; #endif n=read(),m=read(); C[0][0]=1; for (int i=1;i<=n;i++) { C[i][0]=C[i][i]=1; for (int j=1;j<i;j++) C[i][j]=(C[i-1][j-1]+C[i-1][j])%P; } p[0]=1;for (int i=1;i<(n+1)*m;i++) p[i]=(p[i-1]<<1)%P; for (int i=0;i<=n;i++) for (int j=0;j<=n-i&&j<=i;j++) h[i][j][0][0]=1; for (int k=1;k<m;k++) for (int x=0;x<(1<<k);x++) { for (int i=0;i<=n;i++) h[i][0][k][x]=p[i*k]; for (int i=1;i<=n;i++) for (int j=1;j<=n-i&&j<=i;j++) for (int u=0;u<=i;u++) for (int v=0;v<=j;v++) if (u==0&&j==v||i==u&&v==0) inc(h[i][j][k][x],1ll*h[max(u,j-v)][min(u,j-v)][k-1][max(x-(1<<k-1),0)]*h[max(i-u,v)][min(i-u,v)][k-1][max(x-(1<<k-1),0)]%P); else inc(h[i][j][k][x],1ll*C[i][u]*C[j][v]%P*h[max(u,v)][min(u,v)][k-1][x]%P*h[max(i-u,j-v)][min(i-u,j-v)][k-1][x]%P); } for (int i=1;i<=n;i++) for (int j=1;j<=n-i&&j<=i;j++) for (int k=1;k<m;k++) for (int x=1;x<(1<<k);x++) inc(g[i][j][k],h[i][j][k][x]); for (int k=1;k<=m;k++) for (int i=1;i<=n;i++) { inc(f[i][k],f[i][k-1]);inc(f[i][k],f[i][k-1]); for (int j=1;j<i;j++) inc(f[i][k],1ll*C[i][j]*(1ll*f[j][k-1]*p[(i-j)*(k-1)]%P+1ll*f[i-j][k-1]*p[j*(k-1)]%P+p[(k-1)*(i+1)]+g[max(j,i-j)][min(j,i-j)][k-1])%P); } cout<<1ll*f[n][m]*inv(p[m*n])%P; return 0; }
BZOJ4770 圖樣(概率期望+動態規劃)