Submission #1106562

# Submission time Handle Problem Language Result Execution time Memory
1106562 2024-10-30T16:25:44 Z ntdaccode Selling RNA Strands (JOI16_selling_rna) C++17
100 / 100
157 ms 173552 KB
#include<bits/stdc++.h>
#define fori(i,a,b) for(int i=a;i<=b;i++)

using namespace std;
typedef pair<int,int> ii;
typedef tuple<int,int,int> tp;
const int M=1e5+10;
const int N=3e6+10;
const int mod=1e9+7;
int n,m;
int ch[N][4],ch2[N][4];
int digit[26];
int l[N],r[N];
vector<int> suf[N];
int cnt=0,cnt2=0;
void upd(string &s,int &node)
{
  int idx=0;
  int sz=s.size()-1;
  fori(i,0,sz) {
    int k=digit[s[i]-'A'];
    if(!ch[idx][k])
    {
      ch[idx][k]=++cnt;
      l[cnt]=1e9;
      r[cnt]=0;
    }
    idx=ch[idx][k];
    l[idx]=min(l[idx],node);
    r[idx]=max(r[idx],node);
  }

}
void upd2(string &s,int &node)
{
  int idx=0;
  int sz=s.size()-1;
  fori(i,0,sz) {
    int k=digit[s[i]-'A'];
    if(!ch2[idx][k]) ch2[idx][k]=++cnt2;
    idx=ch2[idx][k];
    suf[idx].push_back(node);
  }
}
ii get(string &s)
{
  int idx=0;
  int sz=s.size()-1;
  fori(i,0,sz) {
    int k=digit[s[i]-'A'];
    if(!ch[idx][k]) return {0,0};
    idx=ch[idx][k];
  }
  return {l[idx],r[idx]};
}
int get2(string &s,int &u,int &v)
{
  int idx=0;
  int sz=s.size()-1;
  fori(i,0,sz) {
    int k=digit[s[i]-'A'];
    if(!ch2[idx][k]) return 0;
    idx=ch2[idx][k];
  }
  int uu=lower_bound(suf[idx].begin(),suf[idx].end(),u)-suf[idx].begin();
  int vv=upper_bound(suf[idx].begin(),suf[idx].end(),v)-suf[idx].begin();
  return vv-uu;
}
string s[M];
int32_t main()
{
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);
  if(fopen("1.inp","r"))
  {
    freopen("1.inp","r",stdin);
    freopen("1.out","w",stdout);
  }
  #define task ""
  if(fopen(task".inp","r"))
  {
    freopen(task".inp","r",stdin);
    freopen(task".out","w",stdout);
  }

  cin >> n >> m;
  fori(i,1,n) cin >> s[i];
  digit[0]=0;
  digit[2]=1;
  digit[6]=2;
  digit[20]=3;

  sort(s+1,s+n+1);
  fori(i,1,n){
    upd(s[i],i);
    reverse(s[i].begin(),s[i].end());
    upd2(s[i],i);
  }
  while(m--) {
    string pre,suf;
    cin >> pre >> suf;
    reverse(suf.begin(),suf.end());
    auto [u,v] = get(pre);
    if(u==0&&v==0) cout << 0 << "\n";
    else cout << get2(suf,u,v) << "\n";
  }


}

Compilation message

selling_rna.cpp: In function 'int32_t main()':
selling_rna.cpp:77:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   77 |     freopen("1.inp","r",stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~
selling_rna.cpp:78:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   78 |     freopen("1.out","w",stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~
selling_rna.cpp:83:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   83 |     freopen(task".inp","r",stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
selling_rna.cpp:84:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   84 |     freopen(task".out","w",stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 14 ms 80464 KB Output is correct
2 Correct 15 ms 80480 KB Output is correct
3 Correct 12 ms 80596 KB Output is correct
4 Correct 12 ms 80464 KB Output is correct
5 Correct 13 ms 80464 KB Output is correct
6 Correct 12 ms 80464 KB Output is correct
7 Correct 16 ms 80464 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 135 ms 173552 KB Output is correct
2 Correct 135 ms 168520 KB Output is correct
3 Correct 74 ms 140756 KB Output is correct
4 Correct 76 ms 138936 KB Output is correct
5 Correct 119 ms 166728 KB Output is correct
6 Correct 124 ms 167924 KB Output is correct
7 Correct 48 ms 90700 KB Output is correct
8 Correct 137 ms 137796 KB Output is correct
9 Correct 125 ms 128196 KB Output is correct
10 Correct 98 ms 127192 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 32 ms 81616 KB Output is correct
2 Correct 27 ms 81232 KB Output is correct
3 Correct 29 ms 81232 KB Output is correct
4 Correct 26 ms 81232 KB Output is correct
5 Correct 25 ms 81232 KB Output is correct
6 Correct 31 ms 81340 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 14 ms 80464 KB Output is correct
2 Correct 15 ms 80480 KB Output is correct
3 Correct 12 ms 80596 KB Output is correct
4 Correct 12 ms 80464 KB Output is correct
5 Correct 13 ms 80464 KB Output is correct
6 Correct 12 ms 80464 KB Output is correct
7 Correct 16 ms 80464 KB Output is correct
8 Correct 135 ms 173552 KB Output is correct
9 Correct 135 ms 168520 KB Output is correct
10 Correct 74 ms 140756 KB Output is correct
11 Correct 76 ms 138936 KB Output is correct
12 Correct 119 ms 166728 KB Output is correct
13 Correct 124 ms 167924 KB Output is correct
14 Correct 48 ms 90700 KB Output is correct
15 Correct 137 ms 137796 KB Output is correct
16 Correct 125 ms 128196 KB Output is correct
17 Correct 98 ms 127192 KB Output is correct
18 Correct 32 ms 81616 KB Output is correct
19 Correct 27 ms 81232 KB Output is correct
20 Correct 29 ms 81232 KB Output is correct
21 Correct 26 ms 81232 KB Output is correct
22 Correct 25 ms 81232 KB Output is correct
23 Correct 31 ms 81340 KB Output is correct
24 Correct 157 ms 156748 KB Output is correct
25 Correct 149 ms 156748 KB Output is correct
26 Correct 141 ms 155720 KB Output is correct
27 Correct 80 ms 130192 KB Output is correct
28 Correct 122 ms 100084 KB Output is correct
29 Correct 63 ms 85784 KB Output is correct