Counting Good Teams Gym - 101484K （高維前綴和）

題目

https://vjudge.net/problem/Gym-101484K

題意

給你n個數 問有多少合法數對

定義合法爲 A||B > max(A,B)

思路

高位前綴和 看代碼

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int dp[(1<<21)|5],cnt[(1<<21)|5];
//  子集出現的次數  這個數出現的次數
int main()
{

	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
	{
		int k;
		scanf("%d",&k);
		dp[k]++;cnt[k]++;
	}
	ll ans=1ll*n*(n-1)/2;
	for(int i=0;i<m;i++)
		for(int j=(1<<m)-1;j>0;j--)
			if(j&(1<<i))
				dp[j^(1<<i)]+=dp[j];
	for(int i=0;i<(1<<m);i++)
	{
		ans-=1ll*(dp[i]-cnt[i])*cnt[i]; //(子集-本身)*本身
		ans-=1ll*(cnt[i]-1)*cnt[i]/2;  //自己和自己組合
	}
	printf("%lld\n",ans);
}