●BZOJ 4596 [Shoi2016]黑暗前的幻想鄉


題鏈:

http://www.lydsy.com/JudgeOnline/problem.php?id=4596

題解:

容斥,矩陣樹定理,矩陣行列式


先說說容斥:(一共有 N-1個公司)
令 f[i] 表示選出 (N-1)-i 個公司來修路(即有i個公司一定不修),且不管每個公司只能修一條路這一限制的方案數。
那么 答案 ANS=0個公司不修的方案數 - 1個公司不修的方案數 +2個公司不修的翻案數 ...
即 ANS= f[0] - f[1] +f[2] - ... + (-1)i*f[i]
f[i]的求法呢,就是先O(2N)枚舉公司集合情況,
然后用矩陣樹定理 O(N3) 求出當前情況下的生成樹方案數。
另外再本題中,在構造上三角矩陣用以求行列式時,
既要避免小數,還要不影響原矩陣的行列式的值,
所以采用輾轉相除的高斯消元法去構造上三角矩陣。復雜度多一個(logN)
由矩陣行列式的性質可知,這樣輾轉相除的高斯消元法不會影響行列式的絕對值,
只會影響符號位的正負,所以統計一下正負號的變化就好了。
(還不會矩陣樹定理,看看這里,入門一波。)
所以總的時間復雜度為 O(2N*N3*logN)。
(都是這個復雜度,不曉得為什么我的代碼跑到這么慢,都墊底了......)

代碼:

#include<cstdio>
#include<cstring>
#include<iostream>
#define add(x,y) (((1ll*(x)+(y))%mod+mod)%mod)
#define mul(x,y) (((1ll*(x)*(y))%mod+mod)%mod)
#define filein(x) freopen(#x".in","r",stdin)
#define fileout(x) freopen(#x".out","w",stdout)
using namespace std;
const int mod=1000000007;
struct Matrix{
	int Val[20][20],*X[20],R,C;
	void Init(int r,int c){//r==c
		R=r; C=c;
		memset(Val,0,sizeof(Val));
		for(int i=1;i<=R;i++) X[i]=Val[i];
	}
	void Modify(int r,int c,int v){
		X[r][c]=add(X[r][c],v);
	}
	void operator =(const Matrix &rtm){
		Init(rtm.R,rtm.C);
		for(int i=1;i<=R;i++)
			for(int j=1;j<=C;j++)
				Val[i][j]=rtm.X[i][j];
	}
	Matrix operator -(const Matrix & rtm) const{
		Matrix now; now=*this;
		for(int i=1;i<=R;i++)
			for(int j=1;j<=C;j++)
				now.X[i][j]=add(now.X[i][j],-rtm.X[i][j]);
		return now;
	}
	void Gauss_Euclidean(int p,int &ti){//形成上三角矩陣 
		if(p==R-1) return;
		if(!X[p][p]) 
			for(int i=p+1;i<R;i++) if(X[i][p]){
				swap(X[i],X[p]); ti++; break;
			}
		if(!X[p][p]) return;
		for(int i=p+1;i<R;i++){
			while(X[i][p]){
				int t=X[p][p]/X[i][p];
				for(int j=p;j<R;j++)
					X[p][j]=add(X[p][j],-mul(X[i][j],t));
				swap(X[p],X[i]); ti++;
			}
		}
		Gauss_Euclidean(p+1,ti);
	}
	int Determinant(){
		int ti=0,ans=1;
		Gauss_Euclidean(1,ti);
		for(int i=1;i<R;i++) ans=mul(ans,X[i][i]);
		if(ti&1) ans=mul(ans,-1);
		return ans;
	}
	void print(){
		for(int i=1;(i!=1?printf("\n"):0),i<=R;i++)
			for(int j=1;j<=R;j++)
				printf("%d ",X[i][j]);
	}
}A,B,K;
int cpy[20][500];
int ANS,N,tmp;
void dfs(int p,int num){
	if(p>=N) return;
	//選
	for(int i=1,a,b;i<=2*cpy[p][0];i+=2){
		a=cpy[p][i]; b=cpy[p][i+1];
		A.Modify(a,a,1); A.Modify(b,b,1);
		B.Modify(a,b,1); B.Modify(b,a,1);
	}
	K=A-B; tmp=K.Determinant();
	if(((N-1)-(num+1))&1) tmp=mul(tmp,-1);
	ANS=add(ANS,tmp);
	dfs(p+1,num+1);
	//不選
	for(int i=1,a,b;i<=2*cpy[p][0];i+=2){
		a=cpy[p][i]; b=cpy[p][i+1];
		A.Modify(a,a,-1); A.Modify(b,b,-1);
		B.Modify(a,b,-1); B.Modify(b,a,-1);
	}
	dfs(p+1,num);
}
int main()
{
	scanf("%d",&N);
	A.Init(N,N); B.Init(N,N);
	for(int i=1;i<=N-1;i++){
		scanf("%d",&cpy[i][0]);
		for(int j=1;j<=2*cpy[i][0];j++)
			scanf("%d",&cpy[i][j]);
	}
	dfs(1,0);
	printf("%d",ANS);
	return 0;
}


 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM