【HDOJ】1648 Keywords

PE的注意，如果没有满足条件的不输出空格。
简单模拟，暴力解。

 /*  */
 #include <iostream>
 #include <sstream>
 #include <string>
 #include <map>
 #include <queue>
 #include <set>
 #include <stack>
 #include <vector>
 #include <deque>
 #include <bitset>
 #include <algorithm>
 #include <cstdio>
 #include <cmath>
 #include <ctime>
 #include <cstring>
 #include <climits>
 #include <cctype>
 #include <cassert>
 #include <functional>
 #include <iterator>
 #include <iomanip>
 using namespace std;
 //#pragma comment(linker,"/STACK:102400000,1024000")

 #define sti                set<int>
 #define stpii            set<pair<int, int> >
 #define mpii            map<int,int>
 #define vi                vector<int>
 #define pii                pair<int,int>
 #define vpii            vector<pair<int,int> >
 #define rep(i, a, n)     for (int i=a;i<n;++i)
 #define per(i, a, n)     for (int i=n-1;i>=a;--i)
 #define clr                clear
 #define pb                 push_back
 #define mp                 make_pair
 #define fir                first
 #define sec                second
 #define all(x)             (x).begin(),(x).end()
 #define SZ(x)             ((int)(x).size())
 #define lson            l, mid, rt<<1
 #define rson            mid+1, r, rt<<1|1

 // #define DEBUG

 const int maxp = ;
 const int maxt = ;
 const int maxl = ;
 const int maxn = ;
 const int INF = 0x3f3f3f3f;
 int M[maxn], mn;
 int M_[maxn], mn_;
 map<string,int> tb;
 map<string,int>::iterator iter;
 int wc = , pn = , tn = ;
 char s[maxl];

 typedef struct {
     int l;
     char s[maxl];

     void init() {
         int id;
         int len = strlen(s);
         int i = , j;

         mn = ;
         while (i < len) {
             if (isspace(s[i])) {
                 ++i;
                 continue;
             }
             j = i;
             while (i<len && islower(s[i]))
                 ++i;
             s[i] = '\0';
             #ifdef DEBUG
                 printf("\t%s\n", s+j);
             #endif
             string str(s+j);
             iter = tb.find(str);
             if (iter == tb.end()) {
                 id = tb[str] = wc++;
             } else {
                 id = iter->sec;
             }

             M[mn++] = id;
             s[i] = ' ';
         }

         M[mn] = INF;
     }

 } profile_t;

 typedef struct {
     int a[maxn];
     int sz;
 } tile_t;

 profile_t P[maxp];
 tile_t T[maxt];

 void input_profile() {
     scanf("%d", &P[pn].l);
     gets(P[pn].s);
     // getchar();
     ++pn;
 }

 void input_tile() {
     int l = , id;
     char ch;

     #ifdef DEBUG
         printf("%d:\n", tn);
     #endif

     while () {
         while () {
             ch = getchar();
             if (isspace(ch))
                 continue;
             else
                 break;
         }

         if (ch == '|')
             break;

         // have a word
         l = ;
         while () {
             if (isalpha(ch)) {
                 if (isupper(ch))
                     ch = ch - 'A' + 'a';
                 s[l++] = ch;
             } else if (isspace(ch) || ch=='|') {
                 if (l) {
                     s[l] = '\0';
                     #ifdef DEBUG
                         printf("\t%s\n", s);
                     #endif
                     string str(s);
                     iter = tb.find(str);
                     if (iter == tb.end()) {
                         id = tb[str] = wc++;
                     } else {
                         id = iter->sec;
                     }
                     T[tn].a[T[tn].sz++] = id;
                     l = ;
                 }
                 break;
             }

             ch = getchar();
         }

         if (ch == '|') {
             getchar();
             break;
         }
     }

     ++tn;
 }

 bool judge(int pid, int tid) {
     int l = P[pid].l + ;
     int sz = T[tid].sz;

     // sort(M, M+mn);

     int tmp;
     rep(i, , mn) {
         rep(j, , mn) {
             if (i == j)
                 continue;
             tmp = INF;
             rep(ii, , sz) {
                 if (T[tid].a[ii] != M[i])
                     continue;
                 rep(jj, , sz) {
                     if (jj==ii || T[tid].a[jj]!=M[j])
                         continue;
                     tmp = min(tmp, abs(jj-ii));
                 }
             }

             if (tmp <= l)
                 return true;
         }
     }

     return false;
 }

 void solve_(int pid) {
     int c = ;

     printf("%d:", pid);

     P[pid].init();
     rep(tid, , tn) {
         if (judge(pid, tid)) {
             if (c++)
                 putchar(',');
             else
                 putchar(' ');
             printf("%d", tid);
         }
     }

     putchar('\n');
 }

 void solve() {
     rep(i, , pn)
         solve_(i);
 }

 int main() {
     ios::sync_with_stdio(false);
     #ifndef ONLINE_JUDGE
         freopen("data.in", "r", stdin);
         freopen("data.out", "w", stdout);
     #endif

     char ch;

     while () {
         ch = getchar();
         if (ch == '#')
             break;
         getchar();
         if (ch == 'P') {
             input_profile();
         } else {
             input_tile();
         }
     }

     solve();

     #ifndef ONLINE_JUDGE
         printf("time = %d.\n", (int)clock());
     #endif

     return ;
 }

数据发生器。

 import sys
 import string
 from random import randint

 def GenWords(n = 20):
     ret = []
     lc = list(string.lowercase)
     llc = len(lc) - 1
     for i in xrange(n):
         length = randint(2, 5)
         word = ""
         for j in xrange(length):
             idx = randint(0, llc)
             word += lc[idx]
         ret.append( word )
     return ret

 def GenWord(word):
     op = "!#$%^&*()023+-"
     lop = len(op) - 1
     ret = ""
     for i in xrange(len(word)):
         k = randint(0, 100)
         if k%4 == 0:
             idx = randint(0, lop)
             ret += op[idx]
         if k%4 == 1:
             ret += word[i].upper()
         else:
             ret += word[i]
     return ret

 def GenSpace():
     op = " \t"
     length = randint(1, 3)
     ret = ""
     for i in xrange(length):
         k = randint(0, 1)
         ret += op[k]
     return ret

 def GenTile(wordList):
     lw = len(wordList) - 1
     ret = "T: "
     nline = randint(1, 5)
     lines = []
     for i in xrange(nline):
         nword = randint(2, 10)
         line = ""
         for i in xrange(nword):
             line += GenSpace()
             idx = randint(0, lw)
             line += GenWord(wordList[idx])
         lines.append(line)
     ret = ret + "\n".join(lines) + "|"
     return ret

 def GenPile(wordList):
     lw = len(wordList) - 1
     ret = "P: "
     k = randint(0, 10)
     ret += " %d" % k
     nword = randint(2, 10)
     for i in xrange(nword):
         ret += GenSpace()
         idx = randint(0, lw)
         ret += wordList[idx]
     return ret

 def GenData(fileName):
     wordList = GenWords()
     lw = len(wordList) - 1
     with open(fileName, "w") as fout:
         pn = randint(10, 20)
         tn = randint(10, 30)
         for i in xrange(pn):
             line = GenPile(wordList)
             fout.write("%s\n" % line)
         for i in xrange(tn):
             line = GenTile(wordList)
             fout.write("%s\n" % line)
         fout.write("#\n")

 def MovData(srcFileName, desFileName):
     with open(srcFileName, "r") as fin:
         lines = fin.readlines()
     with open(desFileName, "w") as fout:
         fout.write("".join(lines))

 def CompData():
     print "comp"
     srcFileName = "F:\Qt_prj\hdoj\data.out"
     desFileName = "F:\workspace\cpp_hdoj\data.out"
     srcLines = []
     desLines = []
     with open(srcFileName, "r") as fin:
         srcLines = fin.readlines()
     with open(desFileName, "r") as fin:
         desLines = fin.readlines()
     n = min(len(srcLines), len(desLines))-1
     for i in xrange(n):
         ans2 = int(desLines[i])
         ans1 = int(srcLines[i])
         if ans1 > ans2:
             print "%d: wrong" % i

 if __name__ == "__main__":
     srcFileName = "F:\Qt_prj\hdoj\data.in"
     desFileName = "F:\workspace\cpp_hdoj\data.in"
     GenData(srcFileName)
     MovData(srcFileName, desFileName)