CF917D Stranger Trees

一、题目

点此看题

二、解法

虽然是求生成树，但是真的想不到怎么做啊，可能真就是永远做不出来的题吧。

可以考虑给树边加权$x$，非树边权值固定为$1$，矩阵树能求这个东西：
$$x^k\sum \prod w_e$$不难看出这是一个$n-1$次多项式，它的系数就是我们想要的答案，那么我们可以取$n$个点，就可以解出它的系数表达式。

至于点值计算那一部分，就要用我们的矩阵树定理了。

对于我这个蒟蒻，这道题我可以学的就是可以给特殊边加权

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
#define int long long
const int M = 105;
const int MOD = 1e9+7;
int read()
{
    int num=0,flag=1;
    char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,x[M],y[M],z[M],a[M][M],f[M][M];
int qkpow(int a,int b)
{
    int r=1;
    while(b>0)
    {
        if(b&1) r=r*a%MOD;
        a=a*a%MOD;
        b>>=1;
    }
    return r;
}
int guass()
{
	int ans=1;
    for(int i=2;i<=n;i++)
    {
        for(int j=i+1;j<=n;j++)
            if(!a[i][i] && a[j][i])
            {
                ans=-ans;
                swap(a[i],a[j]);
                break;
            }
        int inv=qkpow(a[i][i],MOD-2);
        for(int j=i+1;j<=n;j++)
        {
            int tmp=a[j][i]*inv%MOD;
            for(int k=i;k<=n;k++)
                a[j][k]=(a[j][k]-a[i][k]*tmp%MOD)%MOD;
        }
    }
    for(int i=2;i<=n;i++)
    	ans=ans*a[i][i]%MOD;
    return (ans+MOD)%MOD;
}
void Guass()
{
	for(int i=1;i<=n;i++)
	{
		for(int j=i+1;j<=n;j++)
			if(!a[i][i] && a[j][i])
			{
				swap(a[i],a[j]);
				break;
			}
		int inv=qkpow(a[i][i],MOD-2);
		for(int j=1;j<=n;j++)
		{
			if(!a[j][i] || i==j) continue;
			int t=a[j][i]*inv%MOD;
			for(int k=i;k<=n+1;k++)
				a[j][k]=(a[j][k]-a[i][k]*t)%MOD;
		}
	}
	for(int i=1;i<=n;i++)
	{
		int t=a[i][n+1]*qkpow(a[i][i],MOD-2)%MOD;
		printf("%lld ",(t+MOD)%MOD);
	}
	puts("");
}
signed main()
{
	n=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read();
		f[u][v]=f[v][u]=1;
	}
	for(int x=0;x<n;x++)
	{
		for(int i=1;i<=n;i++)
			for(int j=i+1;j<=n;j++)
			{
				int v=(f[i][j]?x:1);
				a[i][i]+=v;a[j][j]+=v;
				a[i][j]-=v;a[j][i]-=v;
			}
		z[x]=guass();
		memset(a,0,sizeof a);
	}
	for(int i=1;i<=n;i++)
	{
		int t=1;
		for(int j=1;j<=n;j++)
		{
			a[i][j]=t;
			t=t*(i-1)%MOD;
		}
		a[i][n+1]=z[i-1];
	}
	Guass();
}