当前位置: 代码迷 >> 综合 >> AC自动机+dp打印路径 hdu2296 Ring
  详细解决方案

AC自动机+dp打印路径 hdu2296 Ring

热度:87   发布时间:2023-12-14 03:40:31.0

传送门:点击打开链接

题意:告诉你m个字符串,每个字符串有一个权值。再告诉你n,要求新的字符串长度小于等于n,现在求总权值最大的最小字典序字符串是多少

思路:首先用AC自动机去把状态优化,然后列出dp方程,这些都是不难的,问题还在于打印路径上。由于要求字典序最小,但是实际上在AC自动机顺着插入的话,只能比较后缀,根本比较不了前缀,所以不是很好做。有的题解是插入字符串到AC自动机的时候就把它翻转,这样我就只需要比较后缀就行了,但是比较字符串到时候也需要从后向前去扫,说实话写起来也是很麻烦的。。

这时突然想起了我队友以前经常没事做拿string保存路径递归,,我当时总说这样容易爆内存和超时,但是今天实在想不出什么其他的好办法了,没想到尝试的用了一下string保存路径,不仅代码非常简单,而且效果意外的好,时间上也没有慢太多,感觉有时候这种奇葩思路还是挺不错的。

#include<map>
#include<set>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<cstdio>
#include<cctype>
#include<string>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
#define fuck(x) cout<<"["<<x<<"]"
#define FIN freopen("input.txt","r",stdin)
#define FOUT freopen("output.txt","w+",stdout)
using namespace std;
typedef long long LL;/*MX为总长度*/
const int MN = 100 + 5;
const int MX = 1200 + 5;
const int INF = 0x3f3f3f3f;
const int P = 26;int dp[MN][MX];
string dps[MN][MX];struct AC_machine {int rear, root;int Next[MX][P], Fail[MX], End[MX], ans[MN];void Init() {rear = 0;root = New();}int New() {End[rear] = 0;for(int i = 0; i < P; i++) {Next[rear][i] = -1;}return rear++;}void Add(const char *A, int val) {int now = root, n = strlen(A);for(int i = 0; i < n; i++) {int id = A[i] - 'a';if(Next[now][id] == -1) {Next[now][id] = New();}now = Next[now][id];}End[now] += val;}void Build() {queue<int>Q;Fail[root] = root;for(int i = 0; i < P; i++) {if(Next[root][i] == -1) {Next[root][i] = root;} else {Fail[Next[root][i]] = root;Q.push(Next[root][i]);}}while(!Q.empty()) {int u = Q.front(); Q.pop();End[u] += End[Fail[u]];for(int i = 0; i < P; i++) {if(Next[u][i] == -1) {Next[u][i] = Next[Fail[u]][i];} else {Fail[Next[u][i]] = Next[Fail[u]][i];Q.push(Next[u][i]);}}}}void Solve(int N) {memset(dp, -INF, sizeof(dp));dp[0][0] = 0; dps[0][0] = "";for(int i = 1; i <= N; i++) {for(int j = 0; j < rear; j++) {for(int k = 0; k < P; k++) {int nxt = Next[j][k];int L = dp[i - 1][j] + End[nxt], R = dp[i][nxt];if(L < 0) continue;if(L > R || (L == R && dps[i - 1][j] + (char)('a' + k) < dps[i][nxt])) {dp[i][nxt] = L;dps[i][nxt] = dps[i - 1][j] + (char)('a' + k);}}}}for(int i = 0; i <= N; i++) {ans[i] = 0;for(int j = 1; j < rear; j++) {if(dp[i][j] > dp[i][ans[i]] || (dp[i][j] == dp[i][ans[i]] && dps[i][j] < dps[i][ans[i]])) {ans[i] = j;}}}int len = N;for(int i = N - 1; i >= 0; i--) {if(dp[i][ans[i]] == dp[len][ans[len]]) {len = i;}}printf("%s\n", dps[len][ans[len]].c_str());}
} AC;char word[MX];
vector<string>S;int main() {int T, n, m; //FIN;scanf("%d", &T);while(T--) {AC.Init();S.clear();scanf("%d%d", &n, &m);for(int i = 1; i <= m; i++) {scanf("%s", word);S.push_back(string(word));}for(int i = 0; i < m; i++) {int val;scanf("%d", &val);AC.Add(S[i].c_str(), val);}AC.Build();AC.Solve(n);}return 0;
}