Submission #442050

#	Time	Username	Problem	Language	Result	Execution time	Memory
442050		zaneyu	Selling RNA Strands (JOI16_selling_rna)	C++14	100 / 100	331 ms	276908 KiB

selling_rna

/*input
3 3
AA
AA
AGA
AA AA
AG GA
AG GA
*/
#include<bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
typedef tree<int,null_type,less_equal<int>,rb_tree_tag,tree_order_statistics_node_update> indexed_set;
//order_of_key #of elements less than x
// find_by_order kth element
using ll=long long;
using ld=long double;
using pii=pair<int,int>;
#define f first
#define s second
#define pb push_back
#define REP(i,n) for(int i=0;i<n;i++)
#define REP1(i,n) for(int i=1;i<=n;i++)
#define FILL(n,x) memset(n,x,sizeof(n))
#define ALL(_a) _a.begin(),_a.end()
#define sz(x) (int)x.size()
#define SORT_UNIQUE(c) (sort(c.begin(),c.end()),c.resize(distance(c.begin(),unique(c.begin(),c.end()))))
const ll INF64=1e18+1;
const int INF=0x3f3f3f3f;
int MOD=1e9+7;
const ld PI=acos(-1);
const ld eps=1e-3;
#define lowb(x) x&(-x)
#define MNTO(x,y) x=min(x,(__typeof__(x))y)
#define MXTO(x,y) x=max(x,(__typeof__(x))y)
inline ll mult(ll a,ll b){
    if(a>=MOD) a%=MOD;
    if(b>=MOD) b%=MOD;
    return (a*b)%MOD;
}
inline ll mypow(ll a,ll b){
    if(b<=0) return 1;
    ll res=1LL;
    while(b){
        if(b&1) res=mult(res,a);
        a=mult(a,a);
        b>>=1;
    }
    return res;
}
const int maxn=2e6+5;
const int maxlg=__lg(maxn)+2;
string asd="ACGU";
int get(char x){
    REP(i,4) if(asd[i]==x) return i;
}
struct trie{
    vector<int> v[maxn];
    int arr[maxn][4];
    int cnt=1;
    int search(string s){
        int cur=0;
        REP(i,sz(s)){
            int z=get(s[i]);
            if(!arr[cur][z]){
                return -1;
            }
            cur=arr[cur][z];
        }
        return cur;
    }
    void insert(string s,int idx){
        int cur=0;
        REP(i,sz(s)){
            int z=get(s[i]);
            if(!arr[cur][z]) arr[cur][z]=cnt++;
            cur=arr[cur][z];
            v[cur].pb(idx);
        }
    }
}orig,rev;
string arr[maxn];
int32_t main(){
    ios::sync_with_stdio(false),cin.tie(0);
    int n,q;
    cin>>n>>q;
    REP(i,n) cin>>arr[i];
    sort(arr,arr+n);
    REP(i,n){
        orig.insert(arr[i],i);
        reverse(ALL(arr[i]));
        rev.insert(arr[i],i);
    }
    while(q--){
        string a,b;
        cin>>a>>b;
        reverse(ALL(b));
        int x=orig.search(a),y=rev.search(b);
        if(x==-1 or y==-1){
            cout<<"0\n";
            continue;
        }
        assert(sz(orig.v[x]));
        int l=orig.v[x][0],r=orig.v[x].back();
        cout<<upper_bound(ALL(rev.v[y]),r)-lower_bound(ALL(rev.v[y]),l)<<'\n';
    }
}  

Compilation message (stderr)

selling_rna.cpp: In function 'int get(char)':
selling_rna.cpp:58:1: warning: control reaches end of non-void function [-Wreturn-type]
   58 | }
      | ^

• Subtask 1: 10 / 10
• Subtask 2: 25 / 25
• Subtask 3: 25 / 25
• Subtask 4: 40 / 40