ARC 107 解题报告 A.数学(多项式相乘的性质)B(枚举、计算贡献)C(思维、并查集)D(DP(非整数转化)E(打表找结论)F(网络流不会))

A - Simple Math

题目：
求
$$\sum _{a=1}^A\sum _{b=1}^B\sum _{c=1}^C$$的值

ll a,b,c;
ll inv2;
ll solve(){
	ll res=((a+1)*a%mod*inv2%mod)%mod;
	res=((res*(b+1)%mod)*b%mod*inv2%mod)%mod;
	res=((res*(c+1)%mod)*c%mod*inv2%mod)%mod;
	return res;
}

int main()
{
    read(a),read(b),read(c);
	inv2=ksm(2,mod-2);
	write(solve());
	return 0;
}

B-Quadruple

题目：
给定N、K，求：在1<=a,b,c,d<=N中，有多少对(a,b,c,d)满足a+b-c-d=k
观察上式可知，就是(a+b)-(c+d)=k，记A=(a+b),B=(c+d)，就是求A+B=k的对数，在乘以每对(A,B)对应的(a,b,c,d)对数，所以，预处理A=a+b的对数就行了。难点在于怎么计算对数，通过手写，可以发现，对于一个A，存在的a+b=A的(a,b)的对数的分布有一定的规律。具体看代码。

#define int ll
int n,k;
map<int,int> val;
ll solve(){
	rep(i,2,2*n){
		val[i]=n-abs(n+1-i);//规律在这，建议手动列一下理解。
	}
	ll res=0;
	rep(i,2,2*n){
		res=res+val[i]*val[i-k];
	}
	return res;
}

signed main()
{
	read(n),read(k);
	write(solve());
	return 0;
}

C - Shuffle Permutation

通过手动模拟可以发现，这种可以交换的关系存在传递性，即：a和b可以交换，b和c可以交换，则a和c可以叫交换，所以，并查集维护下集合，并记录下集合元素个数，再组合计数就行。

#define int ll
const int maxn=100+10;
int a[maxn][maxn];
int s1[maxn],s2[maxn];
int siz1[maxn],siz2[maxn];
int f[maxn];
void init(){
	rep(i,1,maxn-1){
		s1[i]=s2[i]=i;
	}
	f[0]=1;
	rep(i,1,maxn-1)f[i]=f[i-1]*i%mod;
}
int find1(int x){
	return s1[x]==x?s1[x]:(s1[x]=find1(s1[x]));
}
int find2(int x){
	return s2[x]==x?s2[x]:(s2[x]=find2(s2[x]));
}
void merge1(int x,int y){
	x=find1(x);
	y=find1(y);
	s1[x]=y;
}
void merge2(int x,int y){
	x=find2(x);
	y=find2(y);
	s2[x]=y;
}
int n,k;
bool check1(int x,int y){
	rep(i,1,n){
		if(a[x][i]+a[y][i]>k) return false;
	}
	return true;
}
bool check2(int x,int y){
	rep(i,1,n){
		if(a[i][x]+a[i][y]>k) return false;
	}
	return true;
}
set<int>ss1,ss2;
ll solve(){
	rep(i,1,n){
		rep(j,i+1,n){
			if(check1(i,j))merge1(i,j);
			if(check2(i,j))merge2(i,j);
		}
	}
	rep(i,1,n){
		siz1[find1(i)]++;
		siz2[find2(i)]++;
	}
	ll res=1;
	rep(i,1,n){
		res=(res*f[siz1[i]])%mod;
		res=(res*f[siz2[i]])%mod;
	}
	return res%mod;
}

signed main()
{
	read(n),read(k);
	init();
	rep(i,1,n)rep(j,1,n) read(a[i][j]);
	write(solve());
	return 0;
}

D - Number of Multisets

非常巧妙的转换。因为原题是1，1/2，1/4，……，我们不可能真的拿分数作为状态递归，但是，我们可以把问题转换一下：我们状态定(i,j)为现在剩余总和为ix，还能用j个数作为状态，而这个x为1，1/2，1/4，……，那么，对于现在剩余总和为ix，若要换用x/2，则i*2维持剩余和不变。注意终止条件。然后记忆化一下。

const int maxn=3e3+100;
int dp[maxn][maxn];
ll solve(int n,int k){
	if(n<=k) return n==k;
	if(k==0) return 0;
	if(~dp[n][k]) return dp[n][k]%mod;
	return dp[n][k]=(solve(n-1,k-1)+solve(n,2*k))%mod;
}

signed main()
{
	int n,k;
	memset(dp,-1,sizeof(dp));
	read(n),read(k);
	write(solve(n,k));
	return 0;
}