正題

題目鏈接:https://ac.nowcoder.com/acm/contest/161/F

---

題目大意

給出$n$個點的一張圖，求它的所有生成樹中權值和為$k$的倍數的個數。輸出答案對$p$取模
$1\le n,k\le 100,1\le m\le 10^4,p\in [2,10^9]\cap Pri$
數據保證$k\equiv 1(modp)$

---

解題思路

一個想法是把一條邊權看做$x^w$的多項式，用矩陣樹定理乘起來后$k$的倍數的系數和就是答案。
但是這樣系數是$nk$個，顯然搞不定。

類似于CF917D-StrangerTree的做法我們可以帶入若干個值然后跑矩陣數之后求出若干個點值。

但是這里是$k$的倍數，我們要模擬卷積，可以帶入$k$個$g$滿足$g^k=1$的就可以了。

這里保證了$k\equiv 1(modp)$，所以我們求出$p$的原根$g$然后帶入$g^{\frac{p?1}{k}\times i}(i\in [0,k?1])$就可以了。

然后直接拉插求出$x=0$的點值就好了。

時間復雜度$O(n^{3}k)$

---

code

#include<cstdio> #include<cstring> #include<algorithm> #include<vector> #define ll long long using namespace std; const ll N=110; struct node{ll x,y,w; }e[N*N]; ll n,m,k,P,g,x[N],y[N]; vector<ll> p; ll power(ll x,ll b){ll ans=1;while(b){if(b&1)ans=ans*x%P;x=x*x%P;b>>=1;}return ans; } void Prime(){ll x=P-1;for(ll i=2;i*i<=x;i++)if(x%i==0){p.push_back(i);while(x%i==0)x/=i;}if(x>1)p.push_back(x); } bool check(ll x){for(ll i=0;i<p.size();i++)if(power(x,(P-1)/p[i])==1)return 0;return 1; } namespace Matrix{ll a[N][N];ll det(ll v){memset(a,0,sizeof(a));ll ans=1;for(ll i=1;i<=m;i++){ll x=e[i].x,y=e[i].y,w=power(v,e[i].w); (a[x][y]+=P-w)%=P;(a[y][x]+=P-w)%=P;(a[x][x]+=w)%=P;(a[y][y]+=w)%=P;}ll f=0;for(ll i=1;i<n;i++){for(ll j=i;j<n;j++)if(a[j][i]){if(i==j)break;swap(a[i],a[j]);f^=1;break;}ans=ans*a[i][i]%P;ll inv=power(a[i][i],P-2);for(ll j=i;j<n;j++)a[i][j]=a[i][j]*inv%P;for(ll j=i+1;j<n;j++){ll rate=P-a[j][i];for(ll k=i;k<n;k++)(a[j][k]+=rate*a[i][k])%=P;}}return f?(P-ans):ans;} } signed main() {scanf("%lld%lld%lld%lld",&n,&m,&k,&P);Prime();g=1;while(!check(g))g++;for(ll i=1;i<=m;i++)scanf("%lld%lld%lld",&e[i].x,&e[i].y,&e[i].w);for(ll i=1;i<=k;i++){x[i]=power(g,(P-1)/k*(i-1));y[i]=Matrix::det(x[i]);}ll ans=0;for(ll i=1;i<=k;i++){ll tmp=1;for(ll j=1;j<=k;j++)if(i!=j)tmp=tmp*(P-x[j])%P*power(x[i]-x[j],P-2)%P;(ans+=tmp*y[i]%P)%=P;}printf("%lld\n",(ans+P)%P);return 0; } 創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的Wannafly挑战赛23F-计数【原根,矩阵树定理,拉格朗日插值】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。