// Palindrome Partitioning #define ROF(i, a, b) for (int i = (b); --i >= (a); ) #define FOR(i, a, b) for (int i = (a); i < (b); i++) #define REP(i, n) for (int i = 0; i < (n); i++) class Solution { private: vector> f; vector rr; vector> r; void g(string &s, int i) { if (i == s.size()) r.push_back(rr); else REP(j, s.size()) if (f[i][j]) { rr.push_back(s.substr(i, j-i+1)); g(s, j+1); rr.pop_back(); } } public: vector> partition(string s) { int n = s.size(); f.assign(n, vector(n)); ROF(i, 0, n) { f[i][i] = true; if (i+1 < n && s[i] == s[i+1]) f[i][i+1] = true; FOR(j, i+2, n) f[i][j] = f[i+1][j-1] && s[i] == s[j]; } r.clear(); g(s, 0); return r; } }; /// #define REP1(i, n) for (remove_cv::type>::type i = 1; i <= (n); i++) class Solution { private: vector r; vector> ret; string s; vector r0, r1; void dfs(int i) { if (i == s.size()) { ret.push_back(r); return; } REP1(j, s.size()-i) if ((j%2 ? r1 : r0)[i+j/2] >= (j+1)/2) { r.push_back(s.substr(i, j)); dfs(i+j); r.pop_back(); } } public: vector> partition(string s) { int n = s.size(); r0.assign(n, 0); r1.assign(n, 0); // r1[i]: 以i爲中心的奇數長度最長迴文子串的長度爲2*r1[i]-1,延伸到i+r1[i]-1 for (int f, g = 0, i = 0; i < n; i++) if (i < g && r1[2*f-i] != g-i) r1[i] = min(r1[2*f-i], g-i); else { f = i; g = max(g, i); for (; g < n && 2*f >= g && s[2*f-g] == s[g]; g++); r1[i] = g-f; } // r0[i]: 以i-1和i和i爲中心的偶數長度最長迴文子串的長度爲2*r0[i],延伸到i+r0[i]-1 for (int f, g = 0, i = 0; i < n; i++) if (i < g && r0[2*f-i] != g-i) r0[i] = min(r0[2*f-i], g-i); else { f = i; g = max(g, i); for (; g < n && 2*f-1 >= g && s[2*f-1-g] == s[g]; g++); r0[i] = g-f; } this->s = std::move(s); ret.clear(); dfs(0); return ret; } };