AcWing 1875. 贝茜的报复(数学+暴力枚举)

题目连接

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

思路

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

代码

#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;
}
上一篇:AGC055C


下一篇:【星河(star)】题解