HDU 1611（I-Keyboard）

#include <cstdio>
#include <cstring>
const int MAXN = 95;

int K, L; //按键数，字符数
char key[MAXN], letter[MAXN]; //按键，字符
int freq[MAXN]; //字符频率
int cost[MAXN][MAXN]; //cost[i][j]表示将第i~j个字符分配到一个按键上的价值
int price[MAXN][MAXN]; //price[i][j]表示将前j个字符分配到前i个按键上的最小价值
int index[MAXN][MAXN]; //第i个按键上分配index[i][j]个字符，且最后一个字符是第j个字符

//初始化
void init()
{
    memset(cost, 0, sizeof(cost));
    memset(price, 0x3f, sizeof(price));
    price[0][0] = 0;
    for (int i = 1; i <= L; i++) //填充cost数组
    {
        for (int j = i; j <= L; j++)
            cost[i][j] = cost[i][j - 1] + (j - i + 1) * freq[j];
    }
}

//动态规划
void dp()
{
    for (int i = 1; i <= K; i++)
    {
        for (int j = i; j <= L; j++)
        {
            for (int m = 1; m <= j - i + 1; m++) //最后一个按键上分配的字符数
            {
                int sum = price[i - 1][j - m] + cost[j - m + 1][j];
                if (sum <= price[i][j])
                {
                    price[i][j] = sum;
                    index[i][j] = m;
                }
            }
        }
    }
}

//递归输出
void output(int K, int L)
{
    if (K == 0) 
        return;
    output(K - 1, L - index[K][L]);
    printf("%c: ", key[K - 1]);
    for (int i = L - index[K][L]; i < L; i++)
    {
        printf("%c", letter[i]);
    }
    printf("\n");
}

int main()
{
    int T;
    scanf("%d", &T);
    for (int Case = 1; Case <= T; Case++)
    {
        scanf("%d%d", &K, &L);
        scanf("%s%s", key, letter);

        memset(freq, 0, sizeof(freq));
        for (int i = 1; i <= L; i++)
        {
            scanf("%d", &freq[i]);
        }

        init();
        dp();

        printf("Keypad #%d:\n", Case);
        output(K, L);
        printf("\n");
    }
    return 0;
}