제출 #464066

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
464066		Jasiekstrz	Trener (COCI20_trener)	C++17	110 / 110	189 ms	18140 KiB

trener

#include<bits/stdc++.h>
#define fi first
#define se second
using namespace std;
const int MOD=1e9+7;
map<string,int> dp;
int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	int n,k;
	cin>>n>>k;
	long long ans=0;
	dp[""]=1;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=k;j++)
		{
			string s;
			cin>>s;
			if(i==n)
				ans=(ans-dp[s])%MOD;
			string s1=s,s2=s;
			s1.pop_back();
			s2.erase(s2.begin());
			dp[s]=((long long)dp[s]+dp[s1]+(s1!=s2)*dp[s2])%MOD;
			if(i==n)
				ans=(ans+dp[s])%MOD;
		}
	}
	cout<<(ans+MOD)%MOD<<"\n";
	return 0;
}

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