Submission #733117

#	Time	Username	Problem	Language	Result	Execution time	Memory
733117		vjudge1	Trener (COCI20_trener)	C++17	110 / 110	91 ms	2512 KiB

trener

#include<bits/stdc++.h>
#define int long long
#define MOD 1000000007
#define all(x) x.begin(),x.end()
#define ff first
#define ss second
#define pb push_back
#define fast ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);

using namespace std;

const int p2=37;

int p[1505];

int add(int x,int y){
	return ((x+y)%MOD+MOD)%MOD;
}

int mul(int x,int y){
	return (x*y)%MOD;
}

int fp(int a,int b){
	if(b==0)return 1;
	int t=fp(a,b/2);
	t=(t*t)%MOD;
	if(b%2)return (t*a)%MOD;
	return t;
}
int32_t main(){
	srand(time(NULL));
	fast;
	p[0]=1;
	for(int i=1;i<=1500;i++)p[i]=mul(p[i-1],p2);
	int n,m;
	cin>>n>>m;
	map<int,int>mp;
	for(int i=0;i<n;i++){
		map<int,int>mp2;
		for(int j=0;j<m;j++){
			string s;
			cin>>s;
			int a=0;
			for(int k=0;k<s.size()-1;k++){
				a=add(a,mul(s[k]-'a'+1,p[k]));
			}
			int b;
			if(i!=0)b=add(mul(s.back()-'a'+1,p[i-1]),mul(fp(p2,MOD-2),add(a,-(s[0]-'a'+1))));
			int c=add(a,mul(s.back()-'a'+1,p[i]));
			if(i==0){
				mp2[c]++;
				mp2[c]%=MOD;
				continue;
			}
			mp2[c]=add(mp2[c],add(mp[a],mp[b]));
			if(a==b){
				mp2[c]=add(mp2[c],-mp[a]);
			}
		}
		mp.clear();
		mp=mp2;
	}
	int ans=0;
	for(auto i:mp)ans=add(ans,i.ss);
	cout<<ans<<endl;
}

Compilation message (stderr)

trener.cpp: In function 'int32_t main()':
trener.cpp:45:17: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   45 |    for(int k=0;k<s.size()-1;k++){
      |                ~^~~~~~~~~~~

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