Submission #88815

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
88815	2018-12-09T01:39:25 Z	jasony123123	Rima (COCI17_rima)	C++11	56 / 140	1000 ms	108244 KB

rima

#define _CRT_SECURE_NO_WARNINGS
#include <bits/stdc++.h>
#include <unordered_map>
//#include <ext/pb_ds/tree_policy.hpp>
//#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
//using namespace __gnu_pbds;
#define FOR(i,start,end) for(int i=start;i<(int)(end);i++)
#define FORE(i,start,end) for(int i=start;i<=(int)end;i++)
#define RFOR(i,start,end) for(int i = start; i>end; i--)
#define RFORE(i,start,end) for(int i = start; i>=end; i--)
#define all(a) a.begin(), a.end()
#define mt make_tuple
#define v vector
#define sf scanf
#define pf printf
#define dvar(x) cout << #x << " := " << x << "\n"
#define darr(x,n) FOR(i,0,n) cout << #x << "[" << i << "]" << " := " << x[i] << "\n"
typedef long long ll;
typedef long double ld;
typedef pair<int, int > pii;
typedef pair<ll, ll> pll;
//template <class T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
template<class T> void minn(T &a, T b) { a = min(a, b); }
template<class T> void maxx(T &a, T b) { a = max(a, b); }
void io() {
#ifdef LOCAL_PROJECT 
	freopen("input.in", "r", stdin); freopen("output.out", "w", stdout);
#else 
	/* online submission */
#endif 
	ios_base::sync_with_stdio(false); cin.tie(NULL);
}
/**************************COCI 2016-2017 R4 P5 *************************/

const int MAXL = 3000000;
const int MAXN = 500000;

struct HashTool {
	const ll MOD = 1000000007LL;
	const ll PRIME = 105943LL;
	static const int SZ = MAXL;
	ll p[SZ];
	HashTool() {
		p[0] = 1;
		FOR(i, 1, SZ) p[i] = (p[i - 1] * PRIME) % MOD;
	}
	void calc(string &str, ll &full, ll &suff) {
		ll hsh = 0;
		RFORE(i, str.size() - 1, 0) {
			hsh = (hsh + ((str[i] * p[str.size() - 1 - i]) % MOD)) % MOD;
			if (i == 1) suff = hsh;
			else if (i == 0) full = hsh;
		}
	}
};

struct StringStuff {
	int len, gid;
	ll fullhsh, suffhsh;
	StringStuff(string &str, HashTool &h) {
		len = str.size();
		h.calc(str, fullhsh, suffhsh);
	}
};


int N;
HashTool ht;
v<StringStuff> vstring;

int NG = 0;
int groupsz[MAXN];
ll groupsuff[MAXN];

map<int, v<int>> allgroups;
map<int, map<ll, int>> allstring; // all hash, group id

v<int> roots;
v<int> chil[MAXN];


int dfs(int x, int f) { // 1== root, 0 = not root
	if (!f) {
		int ans = 0;
		for (int c : chil[x])
			maxx(ans, dfs(c, 0));
		return ans + groupsz[x];
	}
	else {
		pii ans = { 0, 0 };
		for (int c : chil[x]) {
			int res = dfs(c, 0);
			if (res > ans.second)
				swap(ans.second, res);
			if (ans.second > ans.first)
				swap(ans.second, ans.first);
		}
		return ans.first + ans.second + groupsz[x];
	}
}

void printGraph() {
	darr(roots, roots.size());
	FOR(i, 0, NG) {
		cout << "group: " << i << "\n";
		for (int c : chil[i])
			cout << "child: " << c << "\n";
	}
}

void buildGraph() {
	int maxlen = MAXL;
	RFORE(i, maxlen, 1) if (allgroups.find(i) != allgroups.end()) {
		for (int g : allgroups[i]) {
			if (allstring.find(i - 1) == allstring.end())
				roots.push_back(g);
			else {
				auto it = allstring[i - 1].find(groupsuff[g]);
				if (it == allstring[i - 1].end())
					roots.push_back(g);
				else
					chil[it->second].push_back(g);
			}
		}
	}
}

void printCheck() {
	dvar(N);
	FOR(i, 0, N)
		pf("string %d: len = %d    gid = %d    fullhsh = %lld    suffhsh = %lld \n", i, vstring[i].len, vstring[i].gid, vstring[i].fullhsh, vstring[i].suffhsh);
	dvar(NG);
	darr(groupsz, NG);
	darr(groupsuff, NG);

	int maxlen = MAXL;
	RFORE(i, maxlen, 1) {
		if (allgroups.find(i) != allgroups.end()) {
			cout << "at length " << i << "\n";
			for (int g : allgroups[i])
				cout << "group " << g << "\n";
		}
		if (allstring.find(i) != allstring.end()) {
			cout << "at length " << i << "\n";
			for (auto g : allstring[i])
				cout << "allhash " << g.first << " groupid " << g.second << "\n";
		}
	}
}

void genGroups() {
	map<pair<int, ll>, v<int> > same;
	FOR(i, 0, N)
		same[{vstring[i].len, vstring[i].suffhsh}].push_back(i);
	for (auto &group : same) {
		groupsz[NG] = group.second.size();
		groupsuff[NG] = group.first.second;
		allgroups[group.first.first].push_back(NG);
		for (auto &str : group.second) {
			vstring[str].gid = NG;
			allstring[group.first.first][vstring[str].fullhsh] = NG;
		}
		NG++;
	}
}

int main() {
	io();
	cin >> N;
	FOR(i, 0, N) {
		string str;
		cin >> str;
		vstring.push_back(StringStuff(str, ht));
	}

	genGroups();
//	printCheck();

	buildGraph();
//	printGraph();

	int ans = 0;
	for (int r : roots)
		maxx(ans, dfs(r, 1));
	cout << ans << "\n";
	return 0;
}

Subtask #1

56.0 / 140.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	72 ms	35576 KB	Output is correct
2	Correct	59 ms	35772 KB	Output is correct
3	Correct	63 ms	35772 KB	Output is correct
4	Execution timed out	1091 ms	108244 KB	Time limit exceeded
5	Correct	197 ms	108244 KB	Output is correct
6	Incorrect	109 ms	108244 KB	Output isn't correct
7	Incorrect	104 ms	108244 KB	Output isn't correct
8	Incorrect	99 ms	108244 KB	Output isn't correct
9	Incorrect	187 ms	108244 KB	Output isn't correct
10	Incorrect	101 ms	108244 KB	Output isn't correct