hdu 6740 (kmp 枚舉所有可能的循環節維護答案）

題目鏈接

題意：
給你一個小數，已知如果寫全的話，後面爲無限循環小數，現在給你個公式爲a*循環節在這串裏出現的長度-b*循環節的長度，讓這個值最大。

思路：
首先只看小數點之後的數，而且最後一位數字肯定是循環節中的一個，所以我們把小數點後的那一串數倒過來kmp求一下next數組，枚舉每一位i，因爲最小循環節爲i-next[i]， 所以每一位的答案就是a*i-b*(i-next[i])， 維護一下答案就行。

記得要開long long!! （乘法+(數據範圍>1e5) -> longlong）

#include <bits/stdc++.h>

#define eb emplace_back
#define mp make_pair
#define mt make_tuple
#define fi first
#define se second
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define forn(i, n) for (int i = 0; i < (int)(n); ++i)
#define for1(i, n) for (int i = 1; i <= (int)(n); ++i)
#define ford(i, a, b) for (int i = (int)(a); i >= (int)b; --i)
#define fore(i, a, b) for (int i = (int)(a); i <= (int)(b); ++i)
#define rep(i, l, r) for (int i = (l); i <= (r); i++)
#define per(i, r, l) for (int i = (r); i >= (l); i--)
#define ms(x, y) memset(x, y, sizeof(x))
#define SZ(x) ((int)(x).size())

using namespace std;

typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<pii> vpi;
typedef vector<vi> vvi;
typedef long long i64;
typedef vector<i64> vi64;
typedef vector<vi64> vvi64;
typedef pair<i64, i64> pi64;
typedef double ld;

template<class T> bool uin(T &a, T b) { return a > b ? (a = b, true) : false; }
template<class T> bool uax(T &a, T b) { return a < b ? (a = b, true) : false; }

const i64 maxn = (i64)1e7+1000;
i64 nxt[maxn], a, b, ans;
char x[maxn], y[maxn], s[maxn];

void kmp_pre(char x[], i64 m, i64 nxt[]) {
	int i, j;
	j = nxt[0] = -1;
	i = 0;
	while (i <= m) {
		while (-1 != j && x[i] != x[j]) j = nxt[j];
		nxt[++i] = ++j;
	}	
}

int main() {
	ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.precision(10);
    cout << fixed;
#ifdef LOCAL_DEFINE
    freopen("input.txt", "r", stdin);
#endif

     while (cin >> a >> b) {
     	cin >> s;
     	int len = strlen(s);
     	int now = 0;
     	for (int i = len-1; i >= 0; --i) {
     		if (s[i] == '.') break;
     		x[now++] = s[i];
     	}
     	kmp_pre(x, now, nxt);
     	i64 ans = LLONG_MIN;
		for (int i = 1; i <= now; ++i) {
			i64 loop = i-nxt[i];
			ans = max(ans, a*i-b*loop);
		}
		cout << ans << '\n';
     }

#ifdef LOCAL_DEFINE
    cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
	return 0;
}