Submission #297552

# Submission time Handle Problem Language Result Execution time Memory
297552 2020-09-11T15:54:11 Z Hehehe Mate (COCI18_mate) C++14
100 / 100
435 ms 26360 KB
#include<bits/stdc++.h> //:3
#define DIM 2010
#define MOD 1000000007
using namespace std;

//ifstream cin ("date.in");
//ofstream cout ("date.out");
int v[DIM],c[DIM][DIM],f[DIM],dp[DIM][30][30],a[DIM][30];
char s[DIM],l1,l2;
int n,i,j,x,y,q,lg;
int main (){

    scanf("%s",s+1);
    n = strlen(s+1);
    for (i=1;i<=n;i++)
        v[i] = s[i] - 'a' + 1;

    c[0][0] = c[1][0] = c[1][1] = 1;
    for (i=2;i<=n;i++){
        c[i][0] = 1;
        for (j=1;j<=i;j++){
            c[i][j] = c[i-1][j] + c[i-1][j-1];
            if (c[i][j] >= MOD)
                c[i][j] -= MOD;
        }}


    for (i=1;i<=n;i++)
        for (j=i+1;j<=n;j++)
            a[i][v[j]]++;


    for (lg=2;lg<=n;lg++){
        for (i=lg-1;i<=n;i++){
            x = v[i];
            for (y=1;y<=26;y++){
                dp[lg][x][y] = dp[lg][x][y]+1LL*a[i][y]*c[i-1][lg-2]%MOD;
                if (dp[lg][x][y] >= MOD)
                    dp[lg][x][y] -= MOD;
            }}}


    scanf("%d",&q);
    for (;q--;){
        scanf("%d %c %c",&lg,&l1,&l2);
        printf("%d\n",dp[lg][l1-'a'+1][l2-'a'+1]);
    }


    return 0;
}

Compilation message

mate.cpp: In function 'int main()':
mate.cpp:13:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   13 |     scanf("%s",s+1);
      |     ~~~~~^~~~~~~~~~
mate.cpp:43:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   43 |     scanf("%d",&q);
      |     ~~~~~^~~~~~~~~
mate.cpp:45:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   45 |         scanf("%d %c %c",&lg,&l1,&l2);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 9 ms 768 KB Output is correct
2 Correct 6 ms 896 KB Output is correct
3 Correct 7 ms 896 KB Output is correct
4 Correct 11 ms 768 KB Output is correct
5 Correct 47 ms 2808 KB Output is correct
6 Correct 54 ms 2936 KB Output is correct
7 Correct 38 ms 2808 KB Output is correct
8 Correct 35 ms 2680 KB Output is correct
9 Correct 435 ms 26336 KB Output is correct
10 Correct 411 ms 26360 KB Output is correct