oj.uz

Submission #257803

# Submission time Handle Problem Language Result Execution time Memory
257803 2020-08-04T20:20:17 Z Marlov Mate (COCI18_mate) C++14
100 / 100
 684 ms 28408 KB 
/*
Code by @marlov       
*/
#include <iostream>
#include <fstream>
#include <string>
#include <sstream>
#include <vector>
#include <string>
#include <cmath>
#include <algorithm>
#include <iomanip>
#include <utility>
#include <set>
#include <unordered_set>
#include <map>
#include <unordered_map>
#include <stack>
#include <queue>
#include <iterator>
using namespace std;
typedef long long ll;
typedef pair<long long,long long> pi;

#define maxV 2005
#define MOD 1000000007

string s;
long long Q;
long long pc[maxV][maxV];
long long dp[maxV][26];
vector<int> cnts[26];
int main() {
	ios_base::sync_with_stdio(0); cin.tie(0);
	cin>>s;
	cin>>Q;
	//precomute pc[n][m] which is the nCm
	for(long long i=0;i<maxV;i++){
		pc[i][0]=1;
		for(long long j=1;j<i;j++){
			pc[i][j]=pc[i-1][j]+pc[i-1][j-1];
			pc[i][j]%=MOD;
		}
		pc[i][i]=1;
	}
	for(int i=0;i<s.length();i++){
		cnts[s[i]-'a'].push_back(i);
	}
	for(long long i=0;i<26;i++){
		for(long long j=s.length()-1;j>=0;j--){
			if(s[j]=='a'+i){
				dp[j][i]++;
			}
			if(j<s.length()-1){
				dp[j][i]+=dp[j+1][i];
			}
		}
	}
	long long tl;
	char fc,sc;
	for(long long i=0;i<Q;i++){
		long long result=0;
		cin>>tl>>fc>>sc;
		for(long long j=0;j<cnts[fc-'a'].size();j++){
			if(cnts[fc-'a'][j]>=tl-2&&cnts[fc-'a'][j]<s.length()-1){
				result+=pc[cnts[fc-'a'][j]][tl-2]*(dp[cnts[fc-'a'][j]+1][sc-'a']);
				result%=MOD;
			}
		}
		cout<<result<<'\n';
	}
    return 0;
}
/*
malimateodmameitate
3
10 ot
7 aa
3 me
*/
/* stuff you should look for
	* long long overflow, array bounds
	* special cases (n=1,n=0?)
	* do smth instead of nothing and stay organized
*/

Compilation message

mate.cpp: In function 'int main()':
mate.cpp:46:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<s.length();i++){
              ~^~~~~~~~~~~
mate.cpp:54:8: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    if(j<s.length()-1){
       ~^~~~~~~~~~~~~
mate.cpp:64:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(long long j=0;j<cnts[fc-'a'].size();j++){
                     ~^~~~~~~~~~~~~~~~~~~~
mate.cpp:65:45: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    if(cnts[fc-'a'][j]>=tl-2&&cnts[fc-'a'][j]<s.length()-1){

Subtask #1
100.0 / 100.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 23164 KB Output is correct
2 Correct 17 ms 23152 KB Output is correct
3 Correct 20 ms 23168 KB Output is correct
4 Correct 20 ms 23168 KB Output is correct
5 Correct 44 ms 24056 KB Output is correct
6 Correct 47 ms 24184 KB Output is correct
7 Correct 42 ms 23928 KB Output is correct
8 Correct 39 ms 23928 KB Output is correct
9 Correct 650 ms 28408 KB Output is correct
10 Correct 684 ms 28408 KB Output is correct