NBUT 1222 English Game（trie树+DP）

时间：2024-06-27 08:04:19

  • [1222] English Game

  • 时间限制: 1000 ms　内存限制: 131072 K
  • 问题描写叙述

  • This English game is a simple English words connection game.

    The rules are as follows: there are N English words in a dictionary, and every word has its own weight v. There is a weight if the corresponding
    word is used. Now there is a target string X. You have to pick some words in the dictionary, and then connect them to form X. At the same time, the sum weight of the words you picked must be the biggest.

  • 输入
  • There are several test cases. For each test, N (1<=n<=1000) and X (the length of x is not bigger than 10000) are given at first. Then N rows follow. Each row contains a word wi (the length is not bigger than 30) and the weight of it. Every word is composed
    of lowercases. No two words in the dictionary are the same.
  • 输出
  • For each test case, output the biggest sum weight, if you could not form the string X, output -1.
  • 例子输入
  • 1 aaaa
    
a 2
    
3 aaa
    
a 2
    
aa 5
    
aaa 6
    
4 abc
    
a 1
    
bc 2
    
ab 4
    
c 1
    
3 abcd
    
ab 10
    
bc 20
    
cd 30
    
3 abcd
    
cd 100
    
abc 1000
    
bcd 10000
  • 例子输出
  • 8
    
7
    
5
    
40
    
-1

题目大意：

给出一个目标字符串和n个字符串及其相应的权值，求用这些字符串中的1个或多个组成目标字符串的最大权值。

解题思路：

用trie树保存b个字符串及其权值，接下来就是动态规划了。

到达某一点i，遍历trie树找到i+1为第一个字符的字符串。在结束位置j，dp[j]=max(dp[j],dp[i]+val[i+1到j之间的字符串])。除第0位之外，假设dp[i]=0说明没有字符串可以组成0到i之间的字符串，那么就不用查询。

參考代码：

#include<map>

#include<stack>

#include<queue>

#include<cmath>

#include<vector>

#include<cctype>

#include<cstdio>

#include<cstring>

#include<iostream>

#include<algorithm>

using namespace std;

const double eps=1e-10;

const int INF=0x3f3f3f3f;

const int MAXN=1e4+50;

typedef long long LL;

struct node

{

    node* next[26];

    int val;

    node()

    {

        val=0;

        memset(next,0,sizeof(next));

    }

}*root;

int n,L,dp[MAXN];

char s[MAXN];

void trie_insert(node* start,char* word,int v)

{

    node* now=start;

    int l=strlen(word);

    for(int i=0; i<l; i++)

    {

        int id=word[i]-'a';

        if(now->next[id]==NULL)

            now->next[id]=new node();

        now=now->next[id];

    }

    now->val=v;

}

int trie_query(node* start,int pos)

{

    node* now=start;

    int i;

    for(i=pos+1; i<=L; i++)

    {

        int id=s[i]-'a';

        if(now->next[id]==NULL)

            break;

        now=now->next[id];

        if(now->val)

            dp[i]=max(dp[i],dp[pos]+now->val);

    }

}

int main()

{

#ifndef ONLINE_JUDGE

    freopen("in.txt","r",stdin);

#endif // ONLINE_JUDGE

    while(scanf("%d%s",&n,s+1)!=EOF)

    {

        root=new node();

        L=strlen(s+1);

        memset(dp,0,sizeof(dp));

        for(int i=0; i<n; i++)

        {

            char tmp[MAXN];

            int d;

            scanf("%s%d",tmp,&d);

            trie_insert(root,tmp,d);

        }

        for(int i=0; i<=L; i++)

            if(dp[i]||i==0)//假设dp[i]=0说明没有字符串可以组成0到i之间的字符串，那么就不用查询

                trie_query(root,i);

        printf("%d\n",dp[L]==0?-1:dp[L]);

    }

    return 0;

}

相关文章