HDU 4669 Mutiples on a circle （DP , 统计）

转载请注明出处，谢谢http://blog.csdn.net/ACM_cxlove?viewmode=contents    by---cxlove

题意：给出一个环，每个点是一个数字，取一个子串，使得拼接起来的数字是K的倍数。

由于K不大，暂且不考虑环的话，那么dp[i][j]表示以i结尾的，模K为j的有多少个子串。

那么sigma (dp[i][0])便是不考虑环的答案。

考虑环的话，不知道别人怎么写的，我感觉我的写法不是很复杂。

环和情况1 和n肯定是必选的，那么便是一个前缀为后缀，一个后缀为前缀拼接而成。

所以枚举某个前缀，求出前缀模K，那么枚举后缀模K的值，通过之前已经预处理过的dp值，便可以求出有多少个后缀满足为K的倍数。

但是这样可能后缀和前缀重叠了，所以我们枚举前缀的同时，依次记录后缀不同模值的个数。

随着前缀的增长，这些后缀都是和前缀重叠的。

#include <iostream>#include <cstdio>#include <cstring>#include <map>#include <vector>#include <string>#include <queue>#include <cmath>#include <algorithm>#define lson step << 1#define rson step << 1 | 1#pragma comment(linker,"/STACK:102400000,102400000")using namespace std;typedef long long LL;const int N = 50005;const int M = 205;int n , k , a[N] ,l[N];int dp[2][M] , prefix[N] , fac[N << 2] , suffix[N];int cnt[M];int cal (int x) {    int cnt = 0;    while (x) x /= 10 , cnt ++;    return cnt;}int main () {    #ifndef ONLINE_JUDGE        freopen ("input.txt" , "r" , stdin);        // freopen ("output.txt" , "w" , stdout);    #endif    while (scanf ("%d %d" , &n , &k) != EOF) {        fac[0] = 1;        for (int i = 1 ; i <= (n << 2) ; i ++)            fac[i] = fac[i - 1] * 10 % k;        for (int i = 1 ; i <= n ; i ++) {            scanf ("%d" , &a[i]);            l[i] = cal (a[i]);        }        for (int i = 0 ; i < 2 ; i ++) {            for (int j = 0 ; j < k ; j ++)                dp[i][j] = 0;        }        dp[1][a[1] % k] = 1;        LL ans = dp[1][0];        for (int i = 2 ; i <= n ; i ++) {            for (int j = 0 ; j < k ; j ++)                dp[i & 1][j] = 0;            dp[i & 1][a[i] % k] ++;            for (int j = 0 ; j < k ; j ++) {                dp[i & 1][(j * fac[l[i]] + a[i]) %k] += dp[(i - 1) & 1][j];            }            ans += dp[i & 1][0];        }        prefix[0] = 0;suffix[n + 1] = 0;        for (int i = 1 ; i <= n ; i ++) {            prefix[i] = (prefix[i - 1] * fac[l[i]] + a[i]) % k;        }        int len = 0;        for (int i = n ; i >= 1 ; i --) {            suffix[i] = (a[i] * fac[len] + suffix[i + 1]) % k;            len += l[i];        }        len = 0;        for (int i = 0 ; i < k ; i ++)            cnt[i] = 0;        for (int i = 1 ; i <= n ; i ++) {            cnt[suffix[i]] ++;            len += l[i];            int p = prefix[i];            for (int j = 0 ; j < k ; j ++) {                if ((j * fac[len] + p) % k) continue;                ans += dp[n & 1][j] - cnt[j];            }        }        printf ("%I64d\n" , ans);    }    return 0;    }