Codechef SUMCUBE Sum of Cubes 组合、三元环计数

传送门


好久没有做过图论题了……

考虑\(k\)次方的组合意义,实际上,要求的所有方案中导出子图边数的\(k\)次方,等价于有顺序地选出其中\(k\)条边,计算它们在哪一些图中出现过,将所有方案计算出来的答案加起来。

对于\(k\)条边来说,如果它们占据了\(x\)个点,那么它们就会出现在\(2^{n-x}\)张图中。

那么\(k=1\)答案显然是\(m \times 2^{n-2}\)

\(k=2\)时有\(3\)种情况:①两条边重合,等价于\(k=1\);②两条边不重合但共一个顶点,对于一条边\((s,t)\),这种情况的出现的次数为\(d_s + d_t - 2\);③两条边不重合且不共顶点,等于\(M^2\)减上面两种情况的出现次数。

\(k=3\)时情况比较多:①三条边重合,等价于\(k=1\);②其中有一对边重合,等价于\(k=2\);③三元环,使用三元环计数计算;

④三条边共占据\(4\)个点:有\(A \rightarrow B \rightarrow C \rightarrow D\)\(A \rightarrow B , A \rightarrow C , A \rightarrow D\)两种情况。第一种情况可以枚举\(BC\)边,那么出现次数就是\((d_B - 1) \times (d_C - 1)\),注意有可能出现\(AD\)共点形成三元环的情况,所以在最后需要减掉\(3 \times\)三元环个数;第二种情况直接枚举点\(A\)

⑤三条边共占据\(5\)个点,即\(A \rightarrow B \rightarrow C , D \rightarrow E\)的情况。可以枚举\(B\),那么方案数就是\(d_B(d_B - 1)(M - 2)\)。注意在这个时候会有③和④的情况重算,记得减掉。

⑥三条边两两没有共顶点,这个就是$M^3 - $①②③④⑤

三元环计数求一下③这道题就做完了。时间复杂度\(O(M \sqrt{M})\)。注意④⑤中重算部分的系数。

下面的代码里\(k=3\)的情况③④⑤⑥都是算的无序的;可能上面的描述里有序和无序有混淆,麻烦注意。

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

inline int read(){
    int a = 0;
    char c = getchar();
    while(!isdigit(c)) c = getchar();
    while(isdigit(c)){
        a = a * 10 + c - 48;
        c = getchar();
    }
    return a;
}

#define int long long
const int MAXN = 1e5 + 7 , MOD = 1e9 + 7;
int N , M , K , s[MAXN] , t[MAXN] , in[MAXN] , pow2[MAXN];

inline int poww2(int b){return b < 0 ? 0 : pow2[b];}

inline int poww(int a , int b){
	int times = 1;
	while(b){
		if(b & 1) times = times * a % MOD;
		a = a * a % MOD;
		b >>= 1;
	}
	return times;
}

namespace solve1{
    int main(){return poww2(N - 2) * M % MOD;}
}

namespace solve2{
    int main(){
        int ans = solve1::main();
        for(int i = 1 ; i <= M ; ++i){
            ans = (ans + (in[s[i]] + in[t[i]] - 2) * poww2(N - 3) + (M - 1 - (in[s[i]] + in[t[i]] - 2)) * poww2(N - 4)) % MOD;
        }
        return ans;
    }
}

namespace solve3{
	int col[MAXN];
	vector < int > ch[MAXN];

	int calc3(){
		int cnt = 0;
		for(int i = 1 ; i <= M ; ++i)
			if(in[s[i]] < in[t[i]] || (in[s[i]] == in[t[i]] && s[i] < t[i]))
				ch[s[i]].push_back(t[i]);
			else ch[t[i]].push_back(s[i]);
		for(int i = 1 ; i <= N ; ++i){
			for(auto p : ch[i]) col[p] = i;
			for(auto p : ch[i])
				for(auto q : ch[p])
					if(col[q] == i) ++cnt;
		}
		return cnt % MOD;
	}

	int main(){
        int ans = solve2::main() , tmp = solve1::main();
        ans = ((ans - tmp + MOD) * 3 + tmp) % MOD;
		int cnt3 = calc3() , cnt4 = MOD - cnt3 * 3 % MOD , cnt5 = 0 , cnt6 = 0;
		for(int i = 1 ; i <= N ; ++i)
			cnt4 = (cnt4 + in[i] * (in[i] - 1) * (in[i] - 2) / 6) % MOD;
		for(int i = 1 ; i <= M ; ++i)
			cnt4 = (cnt4 + (in[s[i]] - 1) * (in[t[i]] - 1)) % MOD;
		cnt5 = MOD - (cnt4 * 2 + cnt3 * 3) % MOD;
		for(int i = 1 ; i <= N ; ++i)
			cnt5 = (cnt5 + MOD - in[i] * (in[i] - 1) * (in[i] - 2) / 6 % MOD) % MOD;
		for(int i = 1 ; i <= N ; ++i)
			cnt5 = (cnt5 + in[i] * (in[i] - 1) * (M - 2) / 2) % MOD;
		cnt6 = (M * (M - 1) * (M - 2) / 6 - cnt5 - cnt4 - cnt3) % MOD;
        return (ans + 6 * (cnt3 * poww2(N - 3) + cnt4 * poww2(N - 4) + cnt5 * poww2(N - 5) + cnt6 * poww2(N - 6))) % MOD;
    }
}

signed main(){
    pow2[0] = 1;
    for(int i = 1 ; i <= 100000 ; ++i)
        pow2[i] = (pow2[i - 1] << 1) % MOD;
	for(int T = read() ; T ; --T){
		N = read(); M = read(); K = read();
		memset(in , 0 , sizeof(int) * (N + 1));
		memset(solve3::col , 0 , sizeof(int) * (N + 1));
		for(int i = 1 ; i <= N ; ++i)
			solve3::ch[i].clear();
		for(int i = 1 ; i <= M ; ++i){
			s[i] = read(); t[i] = read();
			++in[s[i]]; ++in[t[i]];
		}
		if(K == 1) cout << solve1::main() << endl;
		else if(K == 2) cout << solve2::main() << endl;
		else cout << solve3::main() << endl;
	}
    return 0;
}
posted @ 2019-03-13 19:55  cjoier_Itst  阅读(324)  评论(4编辑  收藏  举报