oj.uz

답안 #403238

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
403238 2021-05-13T00:04:08 Z penguinhacker Lozinke (COCI17_lozinke) C++14
85 / 100
 114 ms 2320 KB 
#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define ar array

#define debug(x) cerr << "[" << #x << "] = [" << x << "]\n"

template<class T> ostream& operator<< (ostream& out, vector<T> v) {
	out << '[';
	for (int i = 0; i < v.size(); ++i) {
		if (i > 0) {
			out << ", ";
		}
		out << v[i];
	}
	return out << ']';
}

template<class T> ostream& operator<< (ostream& out, set<T> v) {
	return out << vector<T>(v.begin(), v.end());
}

template<class A, unsigned int sz> ostream& operator<< (ostream& out, ar<A, sz> a) {
	out << '[';
	for (int i = 0; i < sz; ++i) {
		if (i > 0) out << ", ";
		out << a[i];
	}
	return out << ']';
}

template<class A, class B> ostream& operator<< (ostream& out, pair<A, B> p) {
	return out << '[' << p.first << ", " << p.second << ']';
}

template<class A, class B> ostream& operator<< (ostream& out, map<A, B> mp) {
	return out << vector<pair<A, B>>(mp.begin(), mp.end());
}

const int M=1e9+9, B=101;
int n;
ll ans;
map<string, int> oc[11];
map<int, int> mp;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin >> n;
	for (int i=0; i<n; ++i) {
		string s;
		cin >> s;
		++oc[s.size()][s];
	}
	for (int i=1; i<=10; ++i) {
		for (auto& x : oc[i]) {
			set<int> st;
			string s=x.first;
			int cnt=x.second;
			ans+=(ll)cnt*(cnt-1);
			//cout << s << " " << cnt << endl;
			//debug(s);
			for (int i=0; i<s.size(); ++i) {
				ll hsh=0;
				for (int j=i; j<s.size(); ++j) {
					hsh=(B*hsh+s[j])%M;
					st.insert(hsh);
				}
			}
			//debug(mp);
			//debug(st);
			for (const int& j : st) {
				if (mp.count(j)) {
					ans+=(ll)cnt*mp[j];
					//cout << s << " " << cnt << " " << mp[j] << endl;
				}
			}
			//debug(ans);
			ll hsh=0;
			for (char c : s)
				hsh=(B*hsh+c)%M;
			mp[hsh]+=cnt;
		}
	}
	cout << ans;
	return 0;
}

Compilation message

lozinke.cpp: In function 'int main()':
lozinke.cpp:64:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   64 |    for (int i=0; i<s.size(); ++i) {
      |                  ~^~~~~~~~~
lozinke.cpp:66:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   66 |     for (int j=i; j<s.size(); ++j) {
      |                   ~^~~~~~~~~

Subtask #1
85.0 / 100.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 204 KB Output is correct
3 Correct 2 ms 204 KB Output is correct
4 Correct 1 ms 332 KB Output is correct
5 Correct 5 ms 332 KB Output is correct
6 Correct 7 ms 460 KB Output is correct
7 Correct 9 ms 488 KB Output is correct
8 Correct 7 ms 484 KB Output is correct
9 Correct 40 ms 952 KB Output is correct
10 Incorrect 36 ms 1008 KB Output isn't correct
11 Correct 59 ms 1452 KB Output is correct
12 Correct 72 ms 1476 KB Output is correct
13 Correct 78 ms 1868 KB Output is correct
14 Incorrect 114 ms 2320 KB Output isn't correct
15 Incorrect 80 ms 1944 KB Output isn't correct
16 Correct 18 ms 460 KB Output is correct
17 Correct 3 ms 204 KB Output is correct
18 Correct 4 ms 204 KB Output is correct
19 Correct 77 ms 1812 KB Output is correct
20 Correct 15 ms 516 KB Output is correct