Submission #116865

#	Time	Username	Problem	Language	Result	Execution time	Memory
116865		nhimnam120	Lozinke (COCI17_lozinke)	C++14	100 / 100	424 ms	17344 KiB

lozinke

#include<bits/stdc++.h>
#define fi first
#define se second
using namespace std;
map<string, long long> cnt;
vector<pair<long long,string> > inpp;
set<string> inp;
signed main(){
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	long long n;
	cin >> n;
	for(int i=0;i<n;i++){
		string a;
		cin >> a;
		cnt[a]++;
		inp.insert(a);
	}
	long long ans=0;
	for(auto x: inp){
		inpp.push_back({x.size(),x});
	}
	sort(inpp.begin(),inpp.end());
	for(int i=0;i<inpp.size();i++){
		long long pos=0;
		string a=inpp[i].se;
		ans=ans+(cnt[a]*(cnt[a]-1));
		string c;
		set<string> check;
		for(int j=1;j<inpp[i].fi;j++){
			for(int k=0;k<inpp[i].fi-j+1;k++){
				c=a.substr(k,j);
				check.insert(c);
			}
		}
		for(auto s:check){
			ans=ans+cnt[a]*cnt[s];
		}
	}
	cout << ans;
}

Compilation message (stderr)

lozinke.cpp: In function 'int main()':
lozinke.cpp:24:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<inpp.size();i++){
              ~^~~~~~~~~~~~
lozinke.cpp:25:13: warning: unused variable 'pos' [-Wunused-variable]
   long long pos=0;
             ^~~

• Subtask 1: 100 / 100