Submission #291938

#	Time	Username	Problem	Language	Result	Execution time	Memory
291938		eggag32	Selling RNA Strands (JOI16_selling_rna)	C++17	0 / 100	13 ms	15232 KiB

selling_rna

#pragma GCC optimize ("O3")
#pragma GCC target ("sse4")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
using namespace std;
//typedef long long ll;
typedef long double ld;
typedef vector<int> vi;
typedef pair<int, int> pi;
#define debug(x) cerr << #x << ": " << x << endl
#define debug2(x, y) debug(x), debug(y)
#define repn(i, a, b) for(int i = (int)(a); i < (int)(b); i++)
#define rep(i, a) for(int i = 0; i < (int)(a); i++)
#define all(v) v.begin(), v.end() 
#define mp make_pair
#define pb push_back
#define lb lower_bound
#define ub upper_bound
#define fi first
#define se second
#define sq(x) ((x) * (x))
#define ll int
const int mxN = 1e5 + 5;
const int MOD = 189733;
const int MOD2 = 191161;
 
template<class T> T gcd(T a, T b){ return ((b == 0) ? a : gcd(b, a % b)); }
 
struct HASH{
	size_t operator () (const pair<int, int> &x) const {
		return hash<long long>()(((long long)(x.first)) ^ (((long long)(x.second)) << 32));
	}
};
 
int n, m;
string s[mxN];
pair<pair<string, string>, int> p[mxN];
cc_hash_table<pair<ll, ll>, int, HASH> hshs;
vector<pair<ll, ll>> haS[mxN];
vector<pair<ll, ll>> haP[mxN];
pair<ll, ll> h[mxN];
 
bool cmp(pair<pair<string, string>, int> a, pair<pair<string, string>, int> b){
	return a.fi.fi < b.fi.fi;
}
 
void add(int ind){
	//deletes all the suffix hashes of s[ind]
	ll hsh = 1LL, hsh1 = 1LL;
	for(int j = (int)s[ind].size(); j >= 1; j--){
		hsh = (hsh * 179) + (ll)(s[ind][j - 1] - 'a') + 1;
		hsh1 = (hsh1 * 131) + (ll)(s[ind][j - 1] - 'a') + 1;
		hsh %= MOD;
		hsh1 %= MOD2;
		hshs[mp(hsh, hsh1)]++;
	}
}
 
void del(int ind){
	//adds all the suffix hashes of s[ind]
	ll hsh = 1LL, hsh1 = 1LL;
	for(int j = (int)s[ind].size(); j >= 1; j--){
		hsh = (hsh * 179) + (ll)(s[ind][j - 1] - 'a') + 1;
		hsh1 = (hsh1 * 131) + (ll)(s[ind][j - 1] - 'a') + 1;
		hsh %= MOD;
		hsh1 %= MOD2;
		hshs[mp(hsh, hsh1)]--;
	}
}
 
int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	freopen("input.txt", "r", stdin);
	freopen("output.txt", "w", stdout);
	cin >> n >> m;
	rep(i, n) cin >> s[i];
	sort(s, s + n);
	rep(i, n){
		ll hsh = 1LL, hsh1 = 1LL;
		for(int j = 1; j <= (int)s[i].size(); j++){
			hsh = (hsh * 179) + (ll)(s[i][j - 1] - 'a') + 1;
			hsh1 = (hsh1 * 131) + (ll)(s[i][j - 1] - 'a') + 1;
			hsh %= MOD;
			hsh1 %= MOD2;
			haS[i].pb(mp(hsh, hsh1));
		}
	}
	rep(i, m){
		cin >> p[i].fi.fi >> p[i].fi.se;
		p[i].se = i;
	}
	sort(p, p + m, cmp);
	rep(i, m){
		ll hsh = 1LL, hsh1 = 1LL;
		for(int j = (int)p[i].fi.se.size(); j >= 1; j--){
			hsh = (hsh * 179) + (ll)(p[i].fi.se[j - 1] - 'a') + 1;
			hsh1 = (hsh1 * 131) + (ll)(p[i].fi.se[j - 1] - 'a') + 1;
			hsh %= MOD;
			hsh1 %= MOD2;
		}
		h[i] = mp(hsh, hsh1);
	}
	rep(i, m){
		ll hsh = 1LL, hsh1 = 1LL;
		for(int j = 1; j <= (int)p[i].fi.fi.size(); j++){
			hsh = (hsh * 179) + (ll)(p[i].fi.fi[j - 1] - 'a') + 1;
			hsh1 = (hsh1 * 131) + (ll)(p[i].fi.fi[j - 1] - 'a') + 1;
			hsh %= MOD;
			hsh1 %= MOD2;
			haP[i].pb(mp(hsh, hsh1));
		}
	}
	vi ext(m, 0), ans(m, 0);
	repn(i, 1, m){
		if(p[i].fi.fi.size() >= p[i - 1].fi.fi.size()){
			int f = 1;
			rep(j, p[i - 1].fi.fi.size()){
				if(p[i].fi.fi[j] != p[i - 1].fi.fi[j]){
					f = 0;
					break;
				}
			}
			ext[i] = f;
		}
	}
	int l = 1e9, r = 0;
	int hii = 0;
	rep(i, m){
		if(ext[i]){
			//we would always narrow the [l, r] range
			if(l == 1e9){
				ans[p[i].se] = 0;
				continue;
			}
			//modify the range
			repn(j, l, r + 1){
				if(s[j].size() < p[i].fi.fi.size()){
					del(j);
					l++;
					continue;
				}
				string nw = s[j].substr(0, p[i].fi.fi.size());
				if(haS[j][(int)p[i].fi.fi.size() - 1] == haP[i][(int)p[i].fi.fi.size() - 1]) break;
				del(j);
				l++;
			}
			for(int j = r; j >= l; j--){
				if(s[j].size() < p[i].fi.fi.size()){
					del(j);
					r--;
					continue;
				}
				if(haS[j][(int)p[i].fi.fi.size() - 1] == haP[i][(int)p[i].fi.fi.size() - 1]) break;
				del(j);
				r--;
			}
			if(l > r) l = 1e9;
		}
		else{
			l = 1e9, r = 0;
			hshs.clear();
			int lo = hii, hi = n - 1;
			while(lo < hi){
				int mid = (lo + hi + 1) / 2;
				if(s[mid] < p[i].fi.fi) lo = mid;
				else hi = mid - 1;
			}
			hii = max(hii, lo);
			if(lo == 0 && s[lo] >= p[i].fi.fi) lo = -1;
			int nm = 0;
			repn(j, lo + 1, n){
				if(s[j].size() < p[i].fi.fi.size()) break;
				if(haS[j][(int)p[i].fi.fi.size() - 1] != haP[i][(int)p[i].fi.fi.size() - 1]) break;
				add(j);
				nm += s[j].size();
				l = min(l, j), r = max(r, j);
			}
		}
		ans[p[i].se] = hshs[h[i]];
	}
	rep(i, m) cout << ans[i] << '\n';
	return 0;
}
/*
Things to look out for:
	- Integer overflows
	- Array bounds
	- Special cases
Be careful!
*/

Compilation message (stderr)

selling_rna.cpp: In function 'int main()':
selling_rna.cpp:75:9: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
   75 |  freopen("input.txt", "r", stdin);
      |  ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
selling_rna.cpp:76:9: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
   76 |  freopen("output.txt", "w", stdout);
      |  ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~

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