AcWing 1875. 贝茜的报复（数学+暴力枚举）

Posted 2022-02-16 MangataTS

题目连接

https://www.acwing.com/problem/content/1877/

思路

我们顺着不太好计算，所以我们反着计算，计算出所有满足条件的奇数个数，然后相乘就好了，复杂度为$O(N^4)$,

代码

#include<bits/stdc++.h>
using namespace std;
//----------------自定义部分----------------
#define ll long long
#define int long long
#define mod 1000000007
#define endl "\\n"
#define PII pair<int,int>

int dx[4]=0,-1,0,1,dy[4]=-1,0,1,0;

ll ksm(ll a,ll b) 
	ll ans = 1;
	for(;b;b>>=1LL) 
		if(b & 1) ans = ans * a % mod;
		a = a * a % mod;
	
	return ans;


ll lowbit(ll x)return -x & x;

const int N = 100+10;
//----------------自定义部分----------------
int n,m,q;
vector<int> a[8];
/*
不难发现表达式为(2E+2S+I+B)*(G+O+E+S)*(M+2O)
那么我们只需要求出结果为奇数的可能情况就行，也就是I+B、G+O+E+S、M都为奇数的情况
*/

signed main()

	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	std::cout.tie(nullptr);
	
	string op;
	cin>>n;
	int x;
	for(int i = 1;i <= n; ++i) 
		cin>>op>>x;
		if(op == "B") a[1].push_back(x);
		else if(op == "E") a[2].push_back(x);
		else if(op == "S") a[3].push_back(x);
		else if(op == "I") a[4].push_back(x);
		else if(op == "G") a[5].push_back(x);
		else if(op == "O") a[6].push_back(x);
		else if(op == "M") a[7].push_back(x);
	
	ll ans = 1;
	//计算每个字母的个数，顺便求出总可能数
	for(int i = 1;i <= 7; ++i) a[0].push_back(a[i].size()),ans*=a[i].size();
	
	int cnt[4]=0;
	//这个循环处理的是：B+I为奇数的情况
	for(int i = 0;i < a[0][0]; ++i) 
		for(int j = 0;j < a[0][3]; ++j) 
			if((a[1][i] + a[4][j]) & 1) cnt[1]++;
		
	
	//这个循环处理的是：E+S+G+O为奇数的情况
	for(int i = 0;i < a[0][1]; ++i) //E
		for(int j = 0;j < a[0][2]; ++j) //S
			for(int k = 0;k < a[0][4]; ++k) //G
				for(int l = 0;l < a[0][5]; ++l) //O
					int kk = a[2][i] + a[3][j] + a[5][k] + a[6][l];
					if(kk & 1) cnt[2]++;
				
			
		
	
	//这个循环处理的是M
	for(int j = 0;j < a[0][6]; ++j) //M
		if(a[7][j] & 1) cnt[3]++;
	
	ans -= (cnt[1] * cnt[2] * cnt[3]);
	cout<<ans<<endl;
	
	return 0;