排列组合+prufer序列

P5520 [yLOI2019] 青原樱

简单的插空法
一共有n个位置，m棵树，两棵树之间要有空位

那么，我们把这m棵树以及他们所占的位置拿出来，那道路上是不是还剩下n-m个坑，而这n-m个坑有n-m+1个空位
相当于把m个树苗放到n-m+1个可放的位置上，且每个树苗互不相同

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

ll ans=1,type,n,m,p;

int main()
{
	scanf("%lld%lld%lld%lld",&type,&n,&m,&p);
	
	for(int i=n-m+1;i>n-m+1 -m;i--)  
		ans=(ans*i)%p;
	printf("%lld\n",ans);
	return 0;
}

注意：

P3197 [HNOI2008]越狱

设E为（n个人 m种宗教 怎么放使得相邻的两人宗教不同）的方案数

正难则反，转换为ans=总方案数-E

第一个人有m种选择，第二个人为了不与第一个人不重复，只有m-1个选择，第三个人为了不与第二个人重复，也只有m-1个选择。。。最后总情况有m^n个，那么
E=

#include <iostream>
using namespace std;
typedef long long ll;
const int N=1e6+10,mod=100003;

ll qmi(ll a,ll k,ll p)
{
    ll res=1;
    while(k)
    {
        if(k&1) res=(ll)res*a%p;
        k>>=1;
        a=(ll)a*a%p;
    }
    return res%p;
}

int main()
{

    ll n,m;
    ll ans;
    cin>>m>>n;
 
	ans=qmi(m,n,mod)%mod-m*qmi(m-1,n-1,mod)%mod;
	while(ans<0) ans+=mod; //减法可能会导致负数，C++的取模方式会出锅
	ans=ans%mod;
    cout<<ans<<endl;

    return 0;
}

P2290 [HNOI2004]树的计数

转载TheLostWeak的题解
初识prufer序列
https://www.cnblogs.com/chenxiaoran666/p/prufer.html

n=(int)(input())
if n==1:
	x=(int)(input());
	if x==0:print(1);
	else:print(0);
	exit();
f=[0 for i in range(n+5)];f[0]=1;

for i in range(1,n+1):f[i]=f[i-1]*i;
ans = f[n-2];tot=0;s=input().split();
for i in range(n):
    x=(int)(s[i]);
    if x==0:print(0);exit();
    tot+=x-1;ans//=f[x-1];
if(tot==n-2):print(ans);
else:print(0);
 

P4981 父子

题解：
同样是prufer序列的相关结论的应用

题目中存在一个室长无父亲节点，所以题目要求的是有根树，因此，对于一棵n个结点的无根树，我们有n种选根的可能
ans = n^n-1

#include <iostream>
using namespace std;
typedef long long ll;
const int N=1e6+10,mod=1e9+9;
     
ll qmi(ll a,ll k,ll p)
{
    ll res=1;
    while(k)
    {
        if(k&1) res=(ll)res*a%p;
        k>>=1;
        a=(ll)a*a%p;
    }
    return res%p;
}
int main()
{
	
    ll t,n;
    ll ans;
    cin>>t;
    while(t--)
    {
    	cin>>n;
		ans=qmi(n,n-1,mod);
		cout<<ans%mod<<endl;
	}


    return 0;
}

P2624 [HNOI2008]明明的烦恼

转大佬博客
https://www.cnblogs.com/chenxiaoran666/p/Luogu2624.html

n=(int)(input());k=0;s=0;a=[0 for i in range (n+5)];
for i in range(1,n+1):
    a[i]=(int)(input());
    if a[i]==0:print(0);exit();
    if a[i]!=-1:k+=1;s+=a[i]-1;
if s>n-2:print(0);exit();
f=[0 for i in range(n+5)];f[0]=1;
for i in range(1,n+1):f[i]=f[i-1]*i;
ans=ans=(f[n-2]//f[s]//f[n-2-s])*f[s];
for i in range(1,n+1):
    if a[i]!=-1:ans//=f[a[i]-1];
print(ans*pow(n-k,n-2-s));