題目傳送門

題目大意

給出一個 \(n\) 個點的圖，每個點都有一個權值 \(f_i\) ，如果 \(f_i=-1\) 表示 \(i\) 這個點是壞的。定義一個點是有用的當且僅當它不是壞的，並且它連的點中至少有一個點不是壞的。問有多少種生成樹滿足有用的點的權值之和 \(\le \lim\)。

\(n\le 40\)

思路

其實這道題目不難，只是腦子卡路了而已。。。

可以想到，我們可以先統計選出哪些點為有用點權值 \(\le lim\)，我們記錄有 \(i\) 個有用點且合法的方案數為 \(\text{sum}(i)\)，你發現統計有多少種生成樹滿足有 \(i\) 個有用點其實只跟 \(i\) 有關，這個可以用矩陣樹定理求出來。

具體講一下吧，前面那個東西可以折半搜尋，就是拆成兩部分然後合起來考慮，排個序就好了。時間複雜度 \(\Theta(n2^{n/2})\) 。

後面一個關鍵就在於容斥，你發現其實恰好有 \(i\) 個有用點不是很好求，但是至多有 \(i\) 個有用點其實比較好求，具體來說，連邊的方法就是不壞但不有用的點只能連壞點，壞點可以隨便連，然後求生成樹個數。考慮至多有 \(i\) 個有用點的方案數為 \(F(i)\)，恰好有 \(i\) 個有用點的方案數為 \(G(i)\)，那麼可以得到關係式：

\[G(i)=F(i)-\sum_{j=0}^{i-1}\binom{i}{j}G(j) \]

正確性顯然。此部分時間複雜度瓶頸為矩陣樹定理部分，所以為 \(\Theta(n^3)\)

\(\texttt{Code}\)

#include <bits/stdc++.h>
using namespace std;

#define Int register int
#define mod 1000000007
#define MAXM 1050005
#define MAXN 55

template <typename T> inline void read (T &t){t = 0;char c = getchar();int f = 1;while (c < '0' || c > '9'){if (c == '-') f = -f;c = getchar();}while (c >= '0' && c <= '9'){t = (t << 3) + (t << 1) + c - '0';c = getchar();} t *= f;}
template <typename T,typename ... Args> inline void read (T &t,Args&... args){read (t);read (args...);}
template <typename T> inline void write (T x){if (x < 0){x = -x;putchar ('-');}if (x > 9) write (x / 10);putchar (x % 10 + '0');}

int n,m,lim,tot,cnt1,cnt2,cc[MAXN],bin[MAXN][MAXN],Cnt[MAXN],Sum[MAXN],val[MAXN]; 

int mul (int a,int b){return 1ll * a * b % mod;}
int dec (int a,int b){return a >= b ? a - b : a + mod - b;}
int add (int a,int b){return a + b >= mod ? a + b - mod : a + b;}
int qkpow (int a,int b){int res = 1;for (;b;b >>= 1,a = mul (a,a)) if (b & 1) res = mul (res,a);return res;}
int inv (int x){return qkpow (x,mod - 2);}

struct node{
	int s,d;
	bool operator < (const node &p)const{return s < p.s;}
}c1[MAXM],c2[MAXM];

void dfs1 (int x,int S,int D){
	if (S > lim) return ;
	if (x > m) return c1[++ cnt1] = node {S,D},void ();
	dfs1 (x + 1,S,D);
	if (val[x] != -1) dfs1 (x + 1,S + val[x],D + 1);
}

void dfs2 (int x,int S,int D){
	if (S > lim) return ;
	if (x > n) return c2[++ cnt2] = node {S,D},void ();
	dfs2 (x + 1,S,D);
	if (val[x] != -1) dfs2 (x + 1,S + val[x],D + 1);
}

int mat[MAXN][MAXN];
void Link (int x,int y){mat[x][x] ++,mat[y][y] ++,mat[x][y] --,mat[y][x] --;}

int MatrixTree (int k){//至多k個有用點的方案書 
	for (Int i = 1;i <= n;++ i) for (Int j = 1;j <= n;++ j) mat[i][j] = 0;
	for (Int i = 1;i <= n;++ i) 	
		for (Int j = i + 1;j <= n;++ j)
			if (i <= k){if (j <= k || j > tot) Link (i,j);}
			else if (i > tot) Link (i,j);
			else if (j > tot) Link (i,j);
	for (Int i = 1;i <= n;++ i) for (Int j = 1;j <= n;++ j) mat[i][j] = (mat[i][j] % mod + mod) % mod;
	int res = 1;
	for (Int i = 1;i < n;++ i){
		for (Int j = i + 1;j < n;++ j){
			int t = mul (mat[j][i],inv (mat[i][i]));
			for (Int k = i;k < n;++ k) mat[j][k] = dec (mat[j][k],mul (t,mat[i][k])); 
		}
		res = mul (res,mat[i][i]);
	}
	return res;
}

signed main(){
	read (n,lim),m = (n + 1) / 2;
	for (Int i = 1;i <= n;++ i) read (val[i]),tot += (val[i] != -1);
	dfs1 (1,0,0),dfs2 (m + 1,0,0);
	sort (c1 + 1,c1 + cnt1 + 1);
	sort (c2 + 1,c2 + cnt2 + 1);
	for (Int i = cnt1,j = 1;i;-- i){
		while (j <= cnt2 && c1[i].s + c2[j].s <= lim) cc[c2[j].d] ++,j ++;
		for (Int k = 0;k <= n;++ k) Cnt[c1[i].d + k] = add (Cnt[c1[i].d + k],cc[k]);
	}
	for (Int i = 0;i <= n;++ i) bin[i][0] = 1;
	for (Int i = 1;i <= n;++ i)
		for (Int j = 1;j <= i;++ j)
			bin[i][j] = add (bin[i - 1][j - 1],bin[i - 1][j]); 
	for (Int i = 0;i <= tot;++ i) Sum[i] = MatrixTree (i);
	for (Int i = 1;i <= tot;++ i)
		for (Int j = 0;j < i;++ j)
			Sum[i] = dec (Sum[i],mul (bin[i][j],Sum[j]));
	int res = 0;for (Int i = 0;i <= tot;++ i) res = add (res,mul (Cnt[i],Sum[i]));
	write (res),putchar ('\n');
	return 0; 
}