Submission #1322832

#TimeUsernameProblemLanguageResultExecution timeMemory
1322832dadadadadPIN (CEOI10_pin)C++20
45 / 100
1098 ms79200 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;

void solve(){
	int n, d;
	cin >> n >> d;
	
	vector <str> s(n);
	
	for(int i = 0; i < n; i ++) {
		cin >> s[i];
	}
	
	if(d == 4) {
		unordered_map <str, int> cnt;
		ll ans = 0;
		d --;
		for(int i = 0; i < n; i ++) {
			for(int k = 0; k < 10; k ++) {
				if(s[i][0] == char(k + '0')) continue;
				for(int mas = 0; mas < (1 << 3); mas ++) {
					int mask = mas * 2;
					str s2 = s[i];
					s2[0] = char(k + '0');
					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 k = 0; k < 26; k ++) {
				if(s[i][0] == char(k + 'a')) continue;
				for(int mas = 0; mas < (1 << 3); mas ++) {
					int mask = mas * 2;
					str s2 = s[i];
					s2[0] = char(k + 'a');
					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] ++;
			}
			
		}
		cout << ans;
		return;
	}
	
	unordered_map <str, int> cnt;
	ll 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] ++;
		}
	}
	cout << ans;
}


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...