Artifact 0980b7195d8f96ca1040c3b34564725b5cb76e31:
0000: 23 69 6e 63 6c 75 64 65 20 3c 69 6f 73 74 72 65 #include <iostre
0010: 61 6d 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 73 am>..#include <s
0020: 73 74 72 65 61 6d 3e 0d 0a 23 69 6e 63 6c 75 64 stream>..#includ
0030: 65 20 3c 69 6f 6d 61 6e 69 70 3e 0d 0a 23 69 6e e <iomanip>..#in
0040: 63 6c 75 64 65 20 3c 76 65 63 74 6f 72 3e 0d 0a clude <vector>..
0050: 23 69 6e 63 6c 75 64 65 20 3c 73 74 72 69 6e 67 #include <string
0060: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 6d 61 70 >..#include <map
0070: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 73 65 74 >..#include <set
0080: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 61 6c 67 >..#include <alg
0090: 6f 72 69 74 68 6d 3e 0d 0a 23 69 6e 63 6c 75 64 orithm>..#includ
00a0: 65 20 3c 6e 75 6d 65 72 69 63 3e 0d 0a 23 69 6e e <numeric>..#in
00b0: 63 6c 75 64 65 20 3c 69 74 65 72 61 74 6f 72 3e clude <iterator>
00c0: 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 66 75 6e 63 ..#include <func
00d0: 74 69 6f 6e 61 6c 3e 0d 0a 23 69 6e 63 6c 75 64 tional>..#includ
00e0: 65 20 3c 63 6f 6d 70 6c 65 78 3e 0d 0a 23 69 6e e <complex>..#in
00f0: 63 6c 75 64 65 20 3c 71 75 65 75 65 3e 0d 0a 23 clude <queue>..#
0100: 69 6e 63 6c 75 64 65 20 3c 73 74 61 63 6b 3e 0d include <stack>.
0110: 0a 23 69 6e 63 6c 75 64 65 20 3c 63 6d 61 74 68 .#include <cmath
0120: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 63 61 73 >..#include <cas
0130: 73 65 72 74 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 sert>..#include
0140: 3c 63 73 74 72 69 6e 67 3e 0d 0a 75 73 69 6e 67 <cstring>..using
0150: 20 6e 61 6d 65 73 70 61 63 65 20 73 74 64 3b 0d namespace std;.
0160: 0a 74 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c 6f .typedef long lo
0170: 6e 67 20 4c 4c 3b 0d 0a 74 79 70 65 64 65 66 20 ng LL;..typedef
0180: 63 6f 6d 70 6c 65 78 3c 64 6f 75 62 6c 65 3e 20 complex<double>
0190: 43 4d 50 3b 0d 0a 0d 0a 73 74 72 75 63 74 20 62 CMP;....struct b
01a0: 79 4c 65 6e 67 74 68 20 7b 0d 0a 09 62 6f 6f 6c yLength {...bool
01b0: 20 6f 70 65 72 61 74 6f 72 28 29 28 63 6f 6e 73 operator()(cons
01c0: 74 20 73 74 72 69 6e 67 26 20 6c 68 73 2c 20 63 t string& lhs, c
01d0: 6f 6e 73 74 20 73 74 72 69 6e 67 26 20 72 68 73 onst string& rhs
01e0: 29 20 63 6f 6e 73 74 20 7b 0d 0a 09 09 69 66 28 ) const {....if(
01f0: 6c 68 73 2e 73 69 7a 65 28 29 20 21 3d 20 72 68 lhs.size() != rh
0200: 73 2e 73 69 7a 65 28 29 29 20 72 65 74 75 72 6e s.size()) return
0210: 20 6c 68 73 2e 73 69 7a 65 28 29 20 3c 20 72 68 lhs.size() < rh
0220: 73 2e 73 69 7a 65 28 29 3b 0d 0a 09 09 72 65 74 s.size();....ret
0230: 75 72 6e 20 6c 68 73 20 3c 20 72 68 73 3b 0d 0a urn lhs < rhs;..
0240: 09 7d 0d 0a 7d 3b 0d 0a 63 6c 61 73 73 20 50 72 .}..};..class Pr
0250: 65 66 69 78 46 72 65 65 53 75 62 73 65 74 73 20 efixFreeSubsets
0260: 7b 20 70 75 62 6c 69 63 3a 0d 0a 09 6c 6f 6e 67 { public:...long
0270: 20 6c 6f 6e 67 20 63 61 6e 74 50 72 65 66 46 72 long cantPrefFr
0280: 65 65 53 75 62 73 65 74 73 28 76 65 63 74 6f 72 eeSubsets(vector
0290: 20 3c 73 74 72 69 6e 67 3e 20 77 6f 72 64 73 29 <string> words)
02a0: 20 0d 0a 09 7b 0d 0a 09 09 69 6e 74 20 4e 20 3d ...{....int N =
02b0: 20 77 6f 72 64 73 2e 73 69 7a 65 28 29 3b 0d 0a words.size();..
02c0: 09 09 73 6f 72 74 28 77 6f 72 64 73 2e 62 65 67 ..sort(words.beg
02d0: 69 6e 28 29 2c 20 77 6f 72 64 73 2e 65 6e 64 28 in(), words.end(
02e0: 29 2c 20 62 79 4c 65 6e 67 74 68 28 29 29 3b 0d ), byLength());.
02f0: 0a 0d 0a 09 09 76 65 63 74 6f 72 3c 20 76 65 63 .....vector< vec
0300: 74 6f 72 3c 69 6e 74 3e 20 3e 20 64 63 3b 0d 0a tor<int> > dc;..
0310: 09 09 66 6f 72 28 69 6e 74 20 69 3d 30 3b 20 69 ..for(int i=0; i
0320: 3c 4e 3b 20 2b 2b 69 29 0d 0a 09 09 09 64 63 2e <N; ++i).....dc.
0330: 70 75 73 68 5f 62 61 63 6b 28 20 64 69 72 65 63 push_back( direc
0340: 74 5f 63 68 69 6c 64 72 65 6e 28 77 6f 72 64 73 t_children(words
0350: 5b 69 5d 2c 20 77 6f 72 64 73 2c 20 69 2b 31 29 [i], words, i+1)
0360: 20 29 3b 0d 0a 09 09 64 63 2e 70 75 73 68 5f 62 );....dc.push_b
0370: 61 63 6b 28 20 64 69 72 65 63 74 5f 63 68 69 6c ack( direct_chil
0380: 64 72 65 6e 28 22 22 2c 20 77 6f 72 64 73 2c 20 dren("", words,
0390: 30 29 20 29 3b 0d 0a 09 09 72 65 74 75 72 6e 20 0) );....return
03a0: 72 65 63 28 4e 2c 20 64 63 29 20 2d 20 31 3b 0d rec(N, dc) - 1;.
03b0: 0a 09 7d 0d 0a 0d 0a 09 4c 4c 20 72 65 63 28 69 ..}.....LL rec(i
03c0: 6e 74 20 74 2c 20 76 65 63 74 6f 72 3c 20 76 65 nt t, vector< ve
03d0: 63 74 6f 72 3c 69 6e 74 3e 20 3e 26 20 64 63 20 ctor<int> >& dc
03e0: 29 0d 0a 09 7b 0d 0a 09 09 4c 4c 20 70 20 3d 20 )...{....LL p =
03f0: 31 3b 0d 0a 09 09 66 6f 72 28 69 6e 74 20 69 3d 1;....for(int i=
0400: 30 3b 20 69 3c 64 63 5b 74 5d 2e 73 69 7a 65 28 0; i<dc[t].size(
0410: 29 3b 20 2b 2b 69 29 0d 0a 09 09 09 70 20 2a 3d ); ++i).....p *=
0420: 20 72 65 63 28 64 63 5b 74 5d 5b 69 5d 2c 20 64 rec(dc[t][i], d
0430: 63 29 3b 0d 0a 09 09 72 65 74 75 72 6e 20 70 2b c);....return p+
0440: 31 3b 0d 0a 09 7d 0d 0a 0d 0a 09 62 6f 6f 6c 20 1;...}.....bool
0450: 69 73 5f 70 72 65 66 69 78 28 63 6f 6e 73 74 20 is_prefix(const
0460: 73 74 72 69 6e 67 26 20 70 72 65 2c 20 63 6f 6e string& pre, con
0470: 73 74 20 73 74 72 69 6e 67 26 20 61 6c 6c 29 20 st string& all)
0480: 7b 0d 0a 09 09 72 65 74 75 72 6e 20 65 71 75 61 {....return equa
0490: 6c 28 70 72 65 2e 62 65 67 69 6e 28 29 2c 20 70 l(pre.begin(), p
04a0: 72 65 2e 65 6e 64 28 29 2c 20 61 6c 6c 2e 62 65 re.end(), all.be
04b0: 67 69 6e 28 29 29 3b 0d 0a 09 7d 0d 0a 0d 0a 09 gin());...}.....
04c0: 76 65 63 74 6f 72 3c 69 6e 74 3e 20 64 69 72 65 vector<int> dire
04d0: 63 74 5f 63 68 69 6c 64 72 65 6e 28 63 6f 6e 73 ct_children(cons
04e0: 74 20 73 74 72 69 6e 67 26 20 73 2c 20 76 65 63 t string& s, vec
04f0: 74 6f 72 3c 73 74 72 69 6e 67 3e 26 20 77 2c 20 tor<string>& w,
0500: 69 6e 74 20 69 29 20 7b 0d 0a 09 09 76 65 63 74 int i) {....vect
0510: 6f 72 3c 69 6e 74 3e 20 61 6e 73 3b 0d 0a 09 09 or<int> ans;....
0520: 66 6f 72 28 3b 20 69 3c 77 2e 73 69 7a 65 28 29 for(; i<w.size()
0530: 3b 20 2b 2b 69 29 0d 0a 09 09 09 69 66 28 20 69 ; ++i).....if( i
0540: 73 5f 70 72 65 66 69 78 28 73 2c 77 5b 69 5d 29 s_prefix(s,w[i])
0550: 20 29 7b 0d 0a 09 09 09 09 62 6f 6f 6c 20 64 69 ){......bool di
0560: 72 65 63 74 20 3d 20 74 72 75 65 3b 0d 0a 09 09 rect = true;....
0570: 09 09 66 6f 72 28 69 6e 74 20 6a 3d 30 3b 20 6a ..for(int j=0; j
0580: 3c 61 6e 73 2e 73 69 7a 65 28 29 3b 20 2b 2b 6a <ans.size(); ++j
0590: 29 0d 0a 09 09 09 09 09 64 69 72 65 63 74 20 3d ).......direct =
05a0: 20 64 69 72 65 63 74 20 26 26 20 21 69 73 5f 70 direct && !is_p
05b0: 72 65 66 69 78 28 77 5b 61 6e 73 5b 6a 5d 5d 2c refix(w[ans[j]],
05c0: 20 77 5b 69 5d 29 3b 0d 0a 09 09 09 09 69 66 28 w[i]);......if(
05d0: 20 64 69 72 65 63 74 20 29 0d 0a 09 09 09 09 09 direct ).......
05e0: 61 6e 73 2e 70 75 73 68 5f 62 61 63 6b 28 69 29 ans.push_back(i)
05f0: 3b 0d 0a 09 09 09 7d 0d 0a 09 09 72 65 74 75 72 ;.....}....retur
0600: 6e 20 61 6e 73 3b 0d 0a 09 7d 0d 0a 7d 3b 0d 0a n ans;...}..};..
0610: 0d 0a 2f 2f 20 42 45 47 49 4e 20 43 55 54 20 48 ..// BEGIN CUT H
0620: 45 52 45 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 63 ERE..#include <c
0630: 74 69 6d 65 3e 0d 0a 64 6f 75 62 6c 65 20 73 74 time>..double st
0640: 61 72 74 5f 74 69 6d 65 3b 20 73 74 72 69 6e 67 art_time; string
0650: 20 74 69 6d 65 72 28 29 0d 0a 20 7b 20 6f 73 74 timer().. { ost
0660: 72 69 6e 67 73 74 72 65 61 6d 20 6f 73 3b 20 6f ringstream os; o
0670: 73 20 3c 3c 20 22 20 28 22 20 3c 3c 20 69 6e 74 s << " (" << int
0680: 28 28 63 6c 6f 63 6b 28 29 2d 73 74 61 72 74 5f ((clock()-start_
0690: 74 69 6d 65 29 2f 43 4c 4f 43 4b 53 5f 50 45 52 time)/CLOCKS_PER
06a0: 5f 53 45 43 2a 31 30 30 30 29 20 3c 3c 20 22 20 _SEC*1000) << "
06b0: 6d 73 65 63 29 22 3b 20 72 65 74 75 72 6e 20 6f msec)"; return o
06c0: 73 2e 73 74 72 28 29 3b 20 7d 0d 0a 74 65 6d 70 s.str(); }..temp
06d0: 6c 61 74 65 3c 74 79 70 65 6e 61 6d 65 20 54 3e late<typename T>
06e0: 20 6f 73 74 72 65 61 6d 26 20 6f 70 65 72 61 74 ostream& operat
06f0: 6f 72 3c 3c 28 6f 73 74 72 65 61 6d 26 20 6f 73 or<<(ostream& os
0700: 2c 20 63 6f 6e 73 74 20 76 65 63 74 6f 72 3c 54 , const vector<T
0710: 3e 26 20 76 29 0d 0a 20 7b 20 6f 73 20 3c 3c 20 >& v).. { os <<
0720: 22 7b 20 22 3b 0d 0a 20 20 20 66 6f 72 28 74 79 "{ ";.. for(ty
0730: 70 65 6e 61 6d 65 20 76 65 63 74 6f 72 3c 54 3e pename vector<T>
0740: 3a 3a 63 6f 6e 73 74 5f 69 74 65 72 61 74 6f 72 ::const_iterator
0750: 20 69 74 3d 76 2e 62 65 67 69 6e 28 29 3b 20 69 it=v.begin(); i
0760: 74 21 3d 76 2e 65 6e 64 28 29 3b 20 2b 2b 69 74 t!=v.end(); ++it
0770: 29 0d 0a 20 20 20 6f 73 20 3c 3c 20 27 5c 22 27 ).. os << '\"'
0780: 20 3c 3c 20 2a 69 74 20 3c 3c 20 27 5c 22 27 20 << *it << '\"'
0790: 3c 3c 20 28 69 74 2b 31 3d 3d 76 2e 65 6e 64 28 << (it+1==v.end(
07a0: 29 20 3f 20 22 22 20 3a 20 22 2c 20 22 29 3b 20 ) ? "" : ", ");
07b0: 6f 73 20 3c 3c 20 22 20 7d 22 3b 20 72 65 74 75 os << " }"; retu
07c0: 72 6e 20 6f 73 3b 20 7d 0d 0a 76 6f 69 64 20 76 rn os; }..void v
07d0: 65 72 69 66 79 5f 63 61 73 65 28 63 6f 6e 73 74 erify_case(const
07e0: 20 6c 6f 6e 67 20 6c 6f 6e 67 26 20 45 78 70 65 long long& Expe
07f0: 63 74 65 64 2c 20 63 6f 6e 73 74 20 6c 6f 6e 67 cted, const long
0800: 20 6c 6f 6e 67 26 20 52 65 63 65 69 76 65 64 29 long& Received)
0810: 20 7b 0d 0a 20 62 6f 6f 6c 20 6f 6b 20 3d 20 28 {.. bool ok = (
0820: 45 78 70 65 63 74 65 64 20 3d 3d 20 52 65 63 65 Expected == Rece
0830: 69 76 65 64 29 3b 0d 0a 20 69 66 28 6f 6b 29 20 ived);.. if(ok)
0840: 63 65 72 72 20 3c 3c 20 22 50 41 53 53 45 44 22 cerr << "PASSED"
0850: 20 3c 3c 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 << timer() << e
0860: 6e 64 6c 3b 20 20 65 6c 73 65 20 7b 20 63 65 72 ndl; else { cer
0870: 72 20 3c 3c 20 22 46 41 49 4c 45 44 22 20 3c 3c r << "FAILED" <<
0880: 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c timer() << endl
0890: 3b 0d 0a 20 63 65 72 72 20 3c 3c 20 22 5c 74 6f ;.. cerr << "\to
08a0: 3a 20 5c 22 22 20 3c 3c 20 45 78 70 65 63 74 65 : \"" << Expecte
08b0: 64 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 d << '\"' << end
08c0: 6c 20 3c 3c 20 22 5c 74 78 3a 20 5c 22 22 20 3c l << "\tx: \"" <
08d0: 3c 20 52 65 63 65 69 76 65 64 20 3c 3c 20 27 5c < Received << '\
08e0: 22 27 20 3c 3c 20 65 6e 64 6c 3b 20 7d 20 7d 0d "' << endl; } }.
08f0: 0a 23 64 65 66 69 6e 65 20 43 41 53 45 28 4e 29 .#define CASE(N)
0900: 20 7b 63 65 72 72 20 3c 3c 20 22 54 65 73 74 20 {cerr << "Test
0910: 43 61 73 65 20 23 22 20 3c 3c 20 4e 20 3c 3c 20 Case #" << N <<
0920: 22 2e 2e 2e 22 20 3c 3c 20 66 6c 75 73 68 3b 20 "..." << flush;
0930: 73 74 61 72 74 5f 74 69 6d 65 3d 63 6c 6f 63 6b start_time=clock
0940: 28 29 3b 0d 0a 23 64 65 66 69 6e 65 20 45 4e 44 ();..#define END
0950: 09 20 76 65 72 69 66 79 5f 63 61 73 65 28 5f 2c . verify_case(_,
0960: 20 50 72 65 66 69 78 46 72 65 65 53 75 62 73 65 PrefixFreeSubse
0970: 74 73 28 29 2e 63 61 6e 74 50 72 65 66 46 72 65 ts().cantPrefFre
0980: 65 53 75 62 73 65 74 73 28 77 6f 72 64 73 29 29 eSubsets(words))
0990: 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 29 7b 0d ;}..int main(){.
09a0: 0a 0d 0a 43 41 53 45 28 30 29 0d 0a 09 73 74 72 ...CASE(0)...str
09b0: 69 6e 67 20 77 6f 72 64 73 5f 5b 5d 20 3d 20 7b ing words_[] = {
09c0: 22 68 65 6c 6c 6f 22 2c 22 68 65 6c 6c 22 2c 22 "hello","hell","
09d0: 68 69 22 7d 3b 0d 0a 09 20 20 76 65 63 74 6f 72 hi"};... vector
09e0: 20 3c 73 74 72 69 6e 67 3e 20 77 6f 72 64 73 28 <string> words(
09f0: 77 6f 72 64 73 5f 2c 20 77 6f 72 64 73 5f 2b 73 words_, words_+s
0a00: 69 7a 65 6f 66 28 77 6f 72 64 73 5f 29 2f 73 69 izeof(words_)/si
0a10: 7a 65 6f 66 28 2a 77 6f 72 64 73 5f 29 29 3b 20 zeof(*words_));
0a20: 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 5f 20 3d ...long long _ =
0a30: 20 36 4c 4c 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 6LL; ..END..CAS
0a40: 45 28 31 29 0d 0a 09 73 74 72 69 6e 67 20 77 6f E(1)...string wo
0a50: 72 64 73 5f 5b 5d 20 3d 20 7b 22 61 22 2c 22 62 rds_[] = {"a","b
0a60: 22 2c 22 63 22 2c 22 64 22 7d 3b 0d 0a 09 20 20 ","c","d"};...
0a70: 76 65 63 74 6f 72 20 3c 73 74 72 69 6e 67 3e 20 vector <string>
0a80: 77 6f 72 64 73 28 77 6f 72 64 73 5f 2c 20 77 6f words(words_, wo
0a90: 72 64 73 5f 2b 73 69 7a 65 6f 66 28 77 6f 72 64 rds_+sizeof(word
0aa0: 73 5f 29 2f 73 69 7a 65 6f 66 28 2a 77 6f 72 64 s_)/sizeof(*word
0ab0: 73 5f 29 29 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f s_)); ...long lo
0ac0: 6e 67 20 5f 20 3d 20 31 36 4c 4c 3b 20 0d 0a 45 ng _ = 16LL; ..E
0ad0: 4e 44 0d 0a 43 41 53 45 28 32 29 0d 0a 09 73 74 ND..CASE(2)...st
0ae0: 72 69 6e 67 20 77 6f 72 64 73 5f 5b 5d 20 3d 20 ring words_[] =
0af0: 7b 22 61 22 2c 22 61 62 22 2c 22 61 62 63 22 2c {"a","ab","abc",
0b00: 22 61 62 63 64 22 2c 22 61 62 63 64 65 22 2c 22 "abcd","abcde","
0b10: 61 62 63 64 65 66 22 7d 3b 0d 0a 09 20 20 76 65 abcdef"};... ve
0b20: 63 74 6f 72 20 3c 73 74 72 69 6e 67 3e 20 77 6f ctor <string> wo
0b30: 72 64 73 28 77 6f 72 64 73 5f 2c 20 77 6f 72 64 rds(words_, word
0b40: 73 5f 2b 73 69 7a 65 6f 66 28 77 6f 72 64 73 5f s_+sizeof(words_
0b50: 29 2f 73 69 7a 65 6f 66 28 2a 77 6f 72 64 73 5f )/sizeof(*words_
0b60: 29 29 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 )); ...long long
0b70: 20 5f 20 3d 20 37 4c 4c 3b 20 0d 0a 45 4e 44 0d _ = 7LL; ..END.
0b80: 0a 43 41 53 45 28 33 29 0d 0a 09 73 74 72 69 6e .CASE(3)...strin
0b90: 67 20 77 6f 72 64 73 5f 5b 5d 20 3d 20 7b 22 61 g words_[] = {"a
0ba0: 22 2c 22 62 22 2c 22 61 61 22 2c 22 61 62 22 2c ","b","aa","ab",
0bb0: 22 62 61 22 2c 22 62 62 22 7d 3b 0d 0a 09 20 20 "ba","bb"};...
0bc0: 76 65 63 74 6f 72 20 3c 73 74 72 69 6e 67 3e 20 vector <string>
0bd0: 77 6f 72 64 73 28 77 6f 72 64 73 5f 2c 20 77 6f words(words_, wo
0be0: 72 64 73 5f 2b 73 69 7a 65 6f 66 28 77 6f 72 64 rds_+sizeof(word
0bf0: 73 5f 29 2f 73 69 7a 65 6f 66 28 2a 77 6f 72 64 s_)/sizeof(*word
0c00: 73 5f 29 29 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f s_)); ...long lo
0c10: 6e 67 20 5f 20 3d 20 32 35 4c 4c 3b 20 0d 0a 45 ng _ = 25LL; ..E
0c20: 4e 44 0d 0a 0d 0a 7d 0d 0a 2f 2f 20 45 4e 44 20 ND....}..// END
0c30: 43 55 54 20 48 45 52 45 0d 0a CUT HERE..