【HDOJ】4029 Distinct Sub-matrix

思路是枚举矩阵列数，然后将字符矩阵转换成字符串，通过字符数组求不同子串数目。最后，减去不成立的情况。使用特殊字符分割可能的组合。

  1 /* 4029 */
  2 #include <iostream>
  3 #include <sstream>
  4 #include <string>
  5 #include <map>
  6 #include <queue>
  7 #include <set>
  8 #include <stack>
  9 #include <vector>
 10 #include <deque>
 11 #include <algorithm>
 12 #include <cstdio>
 13 #include <cmath>
 14 #include <ctime>
 15 #include <cstring>
 16 #include <climits>
 17 #include <cctype>
 18 #include <cassert>
 19 #include <functional>
 20 #include <iterator>
 21 #include <iomanip>
 22 using namespace std;
 23 //#pragma comment(linker,"/STACK:102400000,1024000")
 24 
 25 #define sti                set<int>
 26 #define stpii            set<pair<int, int> >
 27 #define mpii            map<int,int>
 28 #define vi                vector<int>
 29 #define pii                pair<int,int>
 30 #define vpii            vector<pair<int,int> >
 31 #define rep(i, a, n)     for (int i=a;i<n;++i)
 32 #define per(i, a, n)     for (int i=n-1;i>=a;--i)
 33 #define clr                clear
 34 #define pb                 push_back
 35 #define mp                 make_pair
 36 #define fir                first
 37 #define sec                second
 38 #define all(x)             (x).begin(),(x).end()
 39 #define SZ(x)             ((int)(x).size())
 40 #define lson            l, mid, rt<<1
 41 #define rson            mid+1, r, rt<<1|1
 42 
 43 const int maxr = 130;
 44 const int maxn = 2e5+5;
 45 const int base = 10007;
 46 map<int,int> tb;
 47 int a[maxn*3];
 48 int height[maxn], rrank[maxn], sa[maxn*3];
 49 int wa[maxn], wb[maxn], wc[maxn], wv[maxn];
 50 char s[maxr][maxr];
 51 int H[maxr][maxr];
 52 int r, c;
 53 
 54 bool c0(int *r, int a, int b) {
 55     return r[a]==r[b] && r[a+1]==r[b+1] && r[a+2]==r[b+2];
 56 }
 57 
 58 bool c12(int k, int *r, int a, int b) {
 59     if (k == 2)
 60         return r[a]<r[b] || (r[a]==r[b] && c12(1, r, a+1, b+1));
 61     else
 62         return r[a]<r[b] || (r[a]==r[b] && wv[a+1]<wv[b+1]);
 63 }
 64 
 65 void sort(int *r, int *a, int *b, int n, int m) {
 66     int i;
 67     
 68     for (i=0; i<n; ++i) wv[i] = r[a[i]];
 69     for (i=0; i<m; ++i) wc[i] = 0;
 70     for (i=0; i<n; ++i) wc[wv[i]]++;
 71     for (i=1; i<m; ++i) wc[i] += wc[i-1];
 72     for (i=n-1; i>=0; --i) b[--wc[wv[i]]] = a[i];
 73 }
 74 
 75 #define F(x) ((x)/3 + ((x)%3==1 ? 0:tb))
 76 #define G(x) ((x)<tb ? (x)*3+1 : ((x)-tb)*3+2)
 77 void dc3(int *r, int *sa, int n, int m) {
 78     int i, j, *rn=r+n, *san=sa+n, ta=0, tb=(n+1)/3, tbc=0, p;
 79     
 80     r[n] = r[n+1] = 0;
 81     for (i=0; i<n; ++i) if (i%3!=0) wa[tbc++] = i;
 82     sort(r+2, wa, wb, tbc, m);
 83     sort(r+1, wb, wa, tbc, m);
 84     sort(r, wa, wb, tbc, m);
 85     for (p=1, rn[F(wb[0])]=0, i=1; i<tbc; ++i)
 86         rn[F(wb[i])] = c0(r, wb[i-1], wb[i]) ? p-1 : p++;
 87     if (p < tbc)
 88         dc3(rn, san, tbc, p);
 89     else
 90         for (i=0; i<tbc; ++i) san[rn[i]] = i;
 91     for (i=0; i<tbc; ++i)
 92         if (san[i] < tb)
 93             wb[ta++] = san[i] * 3;
 94     if (n%3 == 1)
 95         wb[ta++] = n - 1;
 96     sort(r, wb, wa, ta, m);
 97     for (i=0; i<tbc; ++i) wv[wb[i]=G(san[i])] = i;
 98     for (i=0,j=0,p=0; i<ta && j<tbc; ++p)
 99         sa[p] = c12(wb[j]%3, r, wa[i], wb[j]) ? wa[i++] : wb[j++];
100     while (i < ta) sa[p++] = wa[i++];
101     while (j < tbc) sa[p++] = wb[j++];
102 }
103 
104 void calheight(int *r, int *sa, int n) {
105     int i, j, k = 0;
106     
107     for (i=1; i<=n; ++i) rrank[sa[i]] = i;
108     for (i=0; i<n; height[rrank[i++]]=k)
109     for (k?k--:0, j=sa[rrank[i]-1]; r[j+k]==r[i+k]; ++k) ;
110 }
111 
112 void printSa(int n) {
113     for (int i=1; i<=n; ++i)
114         printf("%d ", sa[i]);
115     putchar('\n');
116 }
117 
118 void printHeight(int n) {
119     for (int i=1; i<=n; ++i)
120         printf("%d ", height[i]);
121     putchar('\n');
122 }
123 
124 void solve() {
125     __int64 ans = 0, tot, tmp;
126     int n = 0;
127     
128     memset(H, 0, sizeof(H));
129     rep(w, 1, c+1) {
130         tb.clr();
131         int cn = c - w + 1;
132         int cnt = cn + 1;
133         int sidx = 1;
134         
135         rep(i, 0, r) {
136             rep(j, 0, cn) {
137                 H[i][j] = H[i][j] * base + s[i][j+w-1]-'A';
138                 if (tb.find(H[i][j]) == tb.end())
139                     tb[H[i][j]] = cnt++;
140             }
141         }
142         
143         n = 0;
144         rep(j, 0, cn) {
145             rep(i, 0, r)
146                 a[n++] = tb[H[i][j]];
147             a[n++] = sidx++;
148         }
149         a[n] = 0;
150         
151         dc3(a, sa, n+1, cnt+5);
152         calheight(a, sa, n);
153         
154         tot = 0;
155         rep(i, 1, n+1)
156             tot += (n - sa[i] - height[i]);
157         
158         tmp = n;
159         rep(i, 0, cn) {
160             tmp -= r;
161             tot -= (r+1) * tmp;
162             --tmp;
163         }
164         
165         ans += tot;
166     }
167     
168     printf("%I64d\n", ans);
169 }
170 
171 int main() {
172     ios::sync_with_stdio(false);
173     #ifndef ONLINE_JUDGE
174         freopen("data.in", "r", stdin);
175         freopen("data.out", "w", stdout);
176     #endif
177     
178     int t;
179     
180     scanf("%d", &t);
181     rep(tt, 1, t+1) {
182         scanf("%d %d", &r, &c);
183         rep(i, 0, r)
184             scanf("%s", s[i]);
185         printf("Case #%d: ", tt);
186         solve();
187     }
188     
189     #ifndef ONLINE_JUDGE
190         printf("time = %d.\n", (int)clock());
191     #endif
192     
193     return 0;
194 }

数据生成器。

 1 from random import randint, shuffle
 2 import shutil
 3 import string
 4 
 5 
 6 def GenDataIn():
 7     with open("data.in", "w") as fout:
 8         t = 20
 9         uc = list(string.uppercase)
10         fout.write("%d\n" % (t))
11         for tt in xrange(t):
12             n = randint(20, 50)
13             m = randint(20, 50)
14             fout.write("%d %d\n" % (n, m))
15             for i in xrange(n):
16                 line = ""
17                 for j in xrange(m):
18                     idx = randint(0, 25)
19                     line += uc[idx]
20                 fout.write("%s\n" % (line))
21                 
22                 
23 def MovDataIn():
24     desFileName = "F:\eclipse_prj\workspace\hdoj\data.in"
25     shutil.copyfile("data.in", desFileName)
26 
27     
28 if __name__ == "__main__":
29     GenDataIn()
30     MovDataIn()