Comet OJ

题意

https://www.cometoj.com/contest/52/problem/C?problem_id=2416

思路

这里提供一种容斥的写法（？好像网上没看到这种写法）

题目要求编号为 (i) 的节点不能放在 (p_i) 位置，那我们不妨假设没有这些条件，然后再用二进制容斥的方法减去不满足条件的情况（即固定某些 (i) 在 (p_i) 上，这样会好考虑问题一点）。

然后我们面临的问题就是，计算 (A)（二进制）这些数不能选，(B)（二进制）这些位置不能填的方案数。我们枚举两个数，计算它们的对答案的贡献。设小的数为 (i) ，大的数为 (j) ，下面分四种情况讨论：

• (i,j) 的位置均已确定

  若 (p_i>p_j) ，则造成 ((j-i)(p_i-p_j)(n-cnt_A)!) 的贡献（ (cnt_A) 为 (A) 的二进制中 (1) 的个数）。

• (j) 的位置已经确定

  (i) 能选的位置为 (setminus A)（ (A) 对全集取补），尝试让 (setminus A) 与 (j) 匹配，只有 (setminus A) 中大于 (j) 的数才能得配，匹配结果是这个数减 (j) 。设 (setminus A) 所有数的总匹配结果为 (t) ，它对答案的贡献即为 ((j-i)t(n-cnt_A-1)!)

• (i) 的位置已经确定

  与上一种其实没什么区别，只是 (i) 去匹配 (setminus A) 罢了。

• (i,j) 的位置均未确定

这次是一个集合匹配一个集合，在外面通过情况二来预处理。假设总匹配结果为 (t) ，对答案的贡献即为 ((j-i)t(n-cnt_A-2))

要注意固定数时如果有两个数共用了一个位置（即 (p) 相等），那这个贡献就直接为 (0) 了。

代码

#include<bits/stdc++.h>
#define FOR(i,x,y) for(int i=(x),i##END=(y);i<=i##END;++i)
#define DOR(i,x,y) for(int i=(x),i##END=(y);i>=i##END;--i)
#define lowbit(x) ((x)&-(x))
template<typename T,typename _T>inline bool chk_min(T &x,const _T y){return y<x?x=y,1:0;}
template<typename T,typename _T>inline bool chk_max(T &x,const _T y){return x<y?x=y,1:0;}
typedef long long ll;
int cnt[(1<<16)+5],bin[(1<<16)+5],sum[(1<<16)+5];
int Sum[(1<<16)+5];
ll fac[18];
int n,p[18];

ll f1(int S,int pos)
{
	S=(S|((1<<(pos+1))-1))^((1<<(pos+1))-1);
	return sum[S]-cnt[S]*pos;
}

ll f2(int pos,int S)
{
	S=S&((1<<pos)-1);
	return cnt[S]*pos-sum[S];
}

ll solve(int A,int B)
{
	ll ans=0;
	FOR(i,0,n-1)FOR(j,i+1,n-1)
	{
		if((A>>i&1)&&(A>>j&1))
		{
			if(p[i]>p[j])
				ans+=(j-i)*(p[i]-p[j])*fac[n-cnt[A]];
		}
		else if(!(A>>i&1)&&(A>>j&1))
		{
			ans+=(j-i)*f1(((1<<n)-1)^B,p[j])*fac[n-cnt[A]-1];
		}
		else if((A>>i&1)&&!(A>>j&1))
		{
			ans+=(j-i)*f2(p[i],((1<<n)-1)^B)*fac[n-cnt[A]-1];
		}
		else ans+=(j-i)*Sum[((1<<n)-1)^B]*fac[n-cnt[A]-2];
	}
	return ans;
}
 
int main()
{
	fac[0]=1;FOR(i,1,16)fac[i]=fac[i-1]*i;
	FOR(i,1,1<<16)cnt[i]=cnt[i^lowbit(i)]+1;
	FOR(i,2,1<<16)bin[i]=bin[i>>1]+1;
	FOR(i,1,1<<16)sum[i]=sum[i^lowbit(i)]+bin[lowbit(i)];
	FOR(i,1,1<<16)Sum[i]=Sum[i^lowbit(i)]+f1(i^lowbit(i),bin[lowbit(i)]);
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		FOR(i,0,n-1)scanf("%d",&p[i]),p[i]--;
		ll ans=0;
		FOR(i,0,(1<<n)-1)
		{
			int S=0;
			bool flg=1;
			FOR(j,0,n-1)if(i>>j&1)
			{
				if(S&(1<<p[j]))
				{
					flg=0;
					break;
				}
				S|=1<<p[j];
			}
			if(!flg)continue;
			if(cnt[i]&1)ans-=solve(i,S);
			else ans+=solve(i,S);
		}
		printf("%lld
",ans);
	}
	return 0;
}