Code

2021-11-23 23:34:32 阅读：172 来源： 互联网

#include <stdio.h>
#include <string.h>

#define N 100

/*
这道题的输入要比较小心的处理， 由于字符是按顺序给的，所以我们不需要保存字符

-----------------基本思路------------
n = read()
f[n] = read()

best = 最优权值和

m = read()

循环 m 次:
    读方案， 计算当前方案 cost
    计算有无前缀关系

    if ok:
        print("Yes")
    else:
        print("No")
-------------------------------------
*/


/*
f 是频率， rf 是 f 的副本， 因为在计算最优权值和的时候 f 已经被改变

str 保存编码方案
*/


int f[N], rf[N];
char str[N][N];
int vis[N];
int n, m;


int getMinCost(){
    //所有非叶子结点的和

    int T = n - 1; //合并 n - 1 次
    int cost = 0;

    while(T--){
        int m1, m2; //两个最小的值
        int idx1, idx2; //两个最小值对应的下标

        /*
            每次找最小的两个值 m1, m2
            把 m1 的 vis 标记置 1， m2 标记不动
            然后把 m1 的权值加到 m2 上

            实现合并操作
        */

        //find m1
        m1 = 1e8; //一个很大的数
        for(int i = 1;i <= n;i++){
            if(!vis[i] && f[i] < m1){
                m1 = f[i];
                idx1 = i;
            }
        }
        vis[idx1] = 1;

        //find m2
        m2 = 1e8;
        for(int i = 1;i <= n;i++){
            if(!vis[i] && f[i] < m2){
                m2 = f[i];
                idx2 = i;
            }
        }
        f[idx2] += f[idx1];
        cost += f[idx2];
    }

    return cost;
}

/*
计算两个串是否存在前缀关系
'\0' === 0， 所以可以用 *s 来当判断
*/
int isprefix(char* s, char* t){
    while(*s && *t){
        if(*s != *t) return 0;
        s++;t++;
    }
    return 1;
}
int main(){
    scanf("%d ", &n); //注意加了一个空格， 吸收空白符
    char tmp;
    for(int i = 1;i <= n;i++){
        scanf("%c%d ",&tmp, &f[i]); //注意加了一个空格， 吸收空白符
        rf[i] = f[i];
    }

    int best = getMinCost();

    scanf("%d ", &m); //注意加了一个空格， 吸收空白符

    while(m--){
        int cost = 0;
        for(int i = 1;i <= n;i++){
            scanf("%c%s ", &tmp, str[i]); //注意加了一个空格， 吸收空白符
           
            cost += rf[i] * strlen(str[i]);
        }
        if(cost != best){
            puts("No");
            continue;
        }
        int ok = 1;
        for(int i = 1;i <= n;i++){
            for(int j = i + 1;j <= n;j++){
                if(isprefix(str[i], str[j])){
                    ok = 0;
                }
            }
        }
        if(ok) puts("Yes");
        else puts("No");
    }

}

标签：Code,vis,int,m2,read,m1,权值
来源： https://www.cnblogs.com/usr-void/p/15596175.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

Code