【BZOJ4766】文艺计算姬(prufer序,Matrix-Tree定理)

Description

“奋战三星期,造台计算机”。小W响应号召,花了三星期造了台文艺计算姬。文艺计算姬比普通计算机有更多的艺术细胞。普通计算机能计算一个带标号完全图的生成树个数,而文艺计算姬能计算一个带标号完全二分图的生成树个数。更具体地,给定一个一边点数为n,另一边点数为m,共有n*m条边的带标号完全二分图K_{n,m},计算姬能快速算出其生成树个数。小W不知道计算姬算的对不对,你能帮助他吗?


Solution

关于这题的Matrix-Tree证明看这里:https://blog.sengxian.com/solutions/bzoj-4766

从prufer序的角度考虑:每将一个集合中的点插入prufer序中时,删除的一定是另一个集合中的点,由于最后只剩下不同集合中的两个点,所以prufer一定有 ( n + m 2 ) 个数,两个集合各占 n 1 m 1 个数,答案为 n m 1 m n 1


Code

/************************************************ * Au: Hany01 * Date: May 31st, 2018 * Prob: [BZOJ4766] 文艺计算姬 * Email: hany01@foxmail.com ************************************************/

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<int, int> PII;
#define File(a) freopen(a".in", "r", stdin), freopen(a".out", "w", stdout)

template <typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }
template <typename T> inline bool chkmin(T &a, T b) { return b < a ? a = b, 1 : 0; }

inline int read()
{
    register int _, __; register char c_;
    for (_ = 0, __ = 1, c_ = getchar(); c_ < '0' || c_ > '9'; c_ = getchar()) if (c_ == '-') __ = -1;
    for ( ; c_ >= '0' && c_ <= '9'; c_ = getchar()) _ = (_ << 1) + (_ << 3) + (c_ ^ 48);
    return _ * __;
}

LL p;

inline LL mult(LL a, LL b)
{
    LL Ans = 0;
    for ( ; b; b >>= 1, a = (a + a) % p) if (b & 1) (Ans += a) %= p;
    return Ans;
}

inline LL Pow(LL a, LL b)
{
    LL Ans = 1;
    for ( ; b; b >>= 1, a = mult(a, a)) if (b & 1) Ans = mult(Ans, a);
    return Ans;
}

int main()
{
#ifdef hany01
    File("bzoj4766");
#endif

    LL n, m;

    scanf("%lld%lld%lld", &n, &m, &p);
    printf("%lld\n", mult(Pow(n, m - 1), Pow(m, n - 1)));

    return 0;
}
//秋来相顾尚飘蓬,未就丹砂愧葛洪。
// -- 杜甫《赠李白》
相关文章
相关标签/搜索