Submission #845563

#	Time	Username	Problem	Language	Result	Execution time	Memory
845563		vjudge1	Trener (COCI20_trener)	C++17	0 / 110	10 ms	1376 KiB

trener

#include <bits/stdc++.h>
using namespace std;
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#define int long long
const int mod = 1e9 + 7;
inline long long get1(string const& s) {
    const int p = 29;
    const int m = 1e9 + 7;
    long long hash_value = 0;
    long long p_pow = 1;
    for (char c : s) {
        hash_value = (hash_value + (c - 'a' + 1) * p_pow) % m;
        p_pow = (p_pow * p) % m;
    }
    return hash_value;
}
inline long long get2(string const& s) {
    const int p = 37;
    const int m = 1e9 + 7;
    long long hash_value = 0;
    long long p_pow = 1;
    for (char c : s) {
        hash_value = (hash_value + (c - 'a' + 1) * p_pow) % m;
        p_pow = (p_pow * p) % m;
    }
    return hash_value;
}
inline long long get3(string const& s) {
    const int p = 41;
    const int m = 1e9 + 7;
    long long hash_value = 0;
    long long p_pow = 1;
    for (char c : s) {
        hash_value = (hash_value + (c - 'a' + 1) * p_pow) % m;
        p_pow = (p_pow * p) % m;
    }
    return hash_value;
}
inline array < int , 3 > get(string const& s){
	return {get1(s),get2(s),get3(s)};
}
void solve(){
	int n,k;cin >> n >> k;
	string v[n+1][k];
	map < array < int , 3 > , int > mpa;
	for(int i = 1;i<=n;i++){
		for(int j = 0;j<k;j++){
			cin >> v[i][j];
		}
	}
	for(int i = 0;i<k;i++)mpa[get(v[1][i])]++;
	for(int i = 2;i<=n;i++){
		for(int j = 0;j<k;j++){
			string s1 = string(v[i][j].begin() , v[i][j].end()-1) , s2 = string(v[i][j].begin()+1 , v[i][j].end());
			array < int , 3 > myhash = get(v[i][j]);
			array < int , 3 > childhash0 = get(s1);
			array < int , 3 > childhash1 = get(s2);
			if(childhash0 == childhash1){
				mpa[myhash] = (mpa[myhash] + mpa[childhash0]) % mod;
			}
			else{
				mpa[myhash] = (mpa[myhash] + mpa[childhash0] + mpa[childhash1]) % mod;
			}	
		}
	}
	int ans = 0;
	for(int i = 0;i<k;i++)ans = (ans + mpa[get(v[n][i])]) % mod;
	cout << ans << endl;
}
signed main(){
	ios_base::sync_with_stdio(0);cin.tie(0);
	int testcase = 1;//cin >> testcase;
	while(testcase--)solve();
}

• Subtask 1: 0 / 22
• Subtask 2: 0 / 33
• Subtask 3: 0 / 55