D. Triangle Coloring(组合数学)

Problem - D - Codeforces

给定一个由n个顶点和n条边组成的无向图，其中n能被6整除。每条边都有一个权值，它是一个正整数(大于零)。图有如下结构:它被分成顶点的三元组，第一个三元组由顶点1、2、3组成，第二个三元组由顶点4、5、6组成，依此类推。同一个三元组中的每对顶点都由一条边连接。来自不同三元组的顶点之间没有边。你必须把这个图的顶点画成两种颜色，红色和蓝色。每个顶点应该只有一种颜色，应该有红色顶点和蓝色顶点。如果着色符合这些约束，则称为有效的。颜色的权值是连接两个不同颜色顶点的边的权值之和。设W为有效着色的最大可能权重。计算重量为W的有效颜色的数量，并对其取模打印998244353.输入第一行包含一个整数n (6 < n <3-105, n能被6整除)。第二行包含n个整数w1,2，，n (1 w<1000)边的权重。边1连接顶点1和2，边2连接顶点1和3，边3连接顶点2和3，边4连接顶点4和5，边5连接顶点4和6，边6连接顶点5和6，以此类推。输出打印一个整数——对998244353取模后，具有最大可能权重的有效颜色的数量。例子 

input

Copy

12
1 3 3 7 8 5 2 2 2 2 4 2

output

Copy

36

input

Copy

6
4 2 6 6 6 4

output

Copy

2

题解:顶点个数一定是6的倍数,要想权重最大,一个环内三条边颜色一定不能相同,红色和蓝色顶点都是n/2个,红蓝蓝,红红蓝,各应该是是n/6个,那么染色的情况应该是,

具体一个环该怎么染色分为这几种情况

1.三个权值均不相同,肯定是最大的

2.三个权值两个一样的最大,肯定要选最大的这两条边

3.三个权值两个一样的最小,选一条大的,一条小的,选哪个小的都一样,两种情况

4.三个权值都一样,选哪两条边都一样,所以有三种情况

根据乘法原理结合即可

 

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<vector>
#include<map>
#include<queue>
using namespace std;
#define int long long
const int N = 6e5 + 10;
int mod = 998244353;
int qmi(int a,int k,int p)
{
	int res = 1;
	while(k)
	{
		if(k&1)
		res = res *a%p;
		a = a*a%p;
		k >>= 1;
	}
	return res;
}
int C(int a,int b,int p)
{
	if(b > a)
	return 0;
	int res = 1;
	for(int i = 1,j = a;i <= b;i++,j--)
	{
		res = res*j%p;
		res = res*qmi(i,p-2,p)%p;
	}
	return res;
}
int lucas(int a,int b,int p)
{
	if(a < p && b < p)
	return C(a,b,p);
	return C(a%p,b%p,p)*lucas(a/p,b/p,p)%p;
}
void solve()
{
	int n;
	cin >> n;
	vector<int> a(n+1);
	for(int i = 1;i <= n;i ++)
	cin >> a[i];
	int ans = 1;
	for(int i = 1;i <= n;i++)
	{
		if(i%3 == 0)
		{
			int b[3] = {a[i],a[i-1],a[i-2]};
			sort(b,b+3);
			if(b[0] == b[1]&&b[1] != b[2])
			ans = ans*2%mod;
			if(b[0] == b[1]&&b[2] == b[1])
			ans = ans*3%mod;
		}
	}
	ans = (ans*lucas(n/3,n/6,mod))%mod;
	cout << ans;
}
signed main()
{
	int t = 1;
//	cin >> t;
	while(t--)
	{
		solve();
	} 
}