제출 #1322887

#제출 시각아이디문제언어결과실행 시간메모리
1322887dadadadadPIN (CEOI10_pin)C++20
60 / 100
1099 ms97748 KiB

#include <bits/stdc++.h>
using namespace std;
 
#define ll long long
#define str string
#define fi first
#define se second
#define pll pair<ll, ll>
#define pii pair<int, int>
#define pb push_back
#define all(x) x.begin(), x.end()
 
const ll N = 1e5+5, INF = INT_MAX, MOD = 1e9 + 7;
const ll INFL = LLONG_MAX;

int d14(int n, int d, vector <str> &s) {
	map <str, int> cnt;
	int ans = 0;
	for(int i = 0; i < n; i ++) {
		for(int mask = 0; mask < (1 << 4); mask ++) {
			if(__builtin_popcount(mask) > d) continue;
			str s2 = s[i];
			for(int j = 0; j < 4; j ++) {
				if((mask >> j) & 1) s2[j] = 'X';
			}
			if(__builtin_popcount(mask) % 2 == d % 2) {
				ans += cnt[s2];
			}
			else{
				ans -= cnt[s2];
			}
		}
		for(int mask = 0; mask < (1 << 4); mask ++) {
			str s2 = s[i];
			for(int j = 0; j < 4; j ++) {
				if((mask >> j) & 1) s2[j] = 'X';
			}
			
			cnt[s2] ++;
		}
	}
	return ans;
}

int d2(int n, int d, vector <str> &s) {
	map <str, int> cnt;
	int ans = 0;
	for(int i = 0; i < n; i ++) {
		ans += cnt[s[i]];
		for(int mask = 0; mask < (1 << 4); mask ++) {
			if(__builtin_popcount(mask) != 2) continue;
			int lef = 3, rig = 0;
			while(!((mask >> lef) & 1)) lef --;
			while(!((mask >> rig) & 1)) rig ++;
			
			for(char j = '0'; j <= '9'; j ++) {
				for(char k = '0'; k <= '9'; k ++ ) {
					if(s[i][lef] == j || s[i][rig] == k) continue;
					str s2 = s[i];
					s2[lef] = j;
					s2[rig] = k;
					cnt[s2] ++;
				}
				for(char k = 'a'; k <= 'z'; k ++ ) {
					if(s[i][lef] == j || s[i][rig] == k) continue;
					str s2 = s[i];
					s2[lef] = j;
					s2[rig] = k;
					cnt[s2] ++;
				}
			}
			for(char j = 'a'; j <= 'z'; j ++) {
				for(char k = '0'; k <= '9'; k ++ ) {
					if(s[i][lef] == j || s[i][rig] == k) continue;
					str s2 = s[i];
					s2[lef] = j;
					s2[rig] = k;
					cnt[s2] ++;
				}
				for(char k = 'a'; k <= 'z'; k ++ ) {
					if(s[i][lef] == j || s[i][rig] == k) continue;
					str s2 = s[i];
					s2[lef] = j;
					s2[rig] = k;
					cnt[s2] ++;
				}
			}
		}
	}
	return ans;
}

void solve(){
	int n, d;
	cin >> n >> d;
	
	vector <str> s(n);
	
	for(int i = 0; i < n; i ++) {
		cin >> s[i];
	}
	
	
	if(d == 3) {
		cout << n * (n - 1) / 2 - d2(n, d, s) - d14(n, 1, s) - d14(n, 4, s);
	}
	else{
		if(d == 2) {
			cout << d2(n, d, s);
		}
		else{
			cout << d14(n, d, s);
		}
	}
}


int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	
//	freopen(s".in", "r", stdin); 
//	freopen(s".out", "w", stdout);
	
	
	int t;
	t = 1;
	
	
//	cin >> t;
	for(int i = 1; i <= t; i ++) {
//		cout << "Case " << i << ":\n";
		solve();
//		clean();
	}
	
// 	while(cin >> n){
// 	    solve();
// 	}
	return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...