csustoj p4001 你真的会加法吗?(字典树)

题目链接

http://acm.csust.edu.cn/problem/4001

题目思路

题目要求每个数位上相加不进位,用模k后值代替 ,并且数字长度最长为10位。
我们考虑用字典树求解。
因为数字含有前导零,并且我们知道他的最大长度。
我们将数字看作一个字符串,长度不够十位,就在数字前面补0。
之后我们将这个字符串插入字典树。
然后我们查询的方式类似01字典树。
也是先把数字变成长度为十的字符串,然后放入字典树查找。
从第一位开始比较,应为我们要找的是最大值,所以我们从k-该为数字值开始想下找,如果能够走就走,不行就继续向下。当我们走到0时,跳转到k-1,继续找。举个例子:当前数是3,我们的模数是7,当字典树结点上值位3时,该位值会最大,如果这个点存在,那么走向该点,否则判断2再判断1、0。如果都不行,我们就跳转到6的位置,再继续判断,一直到4,跑完一圈后一定会有一个值可以走。
因为存放方法的原因,我们最后一起输出并不好输出,所以直接在查找时输出值。
输出的时候要注意前导零不输出还有需要注意的就是输出0的情况。

ac代码

#pragma comment(linker, "/STACK:102400000,102400000")
#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <utility>
#define pi 3.1415926535898
#define ll long long
#define lson rt<<1
#define rson rt<<1|1h
#define eps 1e-6
#define ms(a,b) memset(a,b,sizeof(a))
#define legal(a,b) a&b
#define print1 printf("111\n")
#define pb(x) push_back(x)
using namespace std;
const int maxn = 2e6+10;
const int inf = 0x3f3f3f3f;
const ll llinf =1e17+10;
const ll mod = 1e9+7;


int a[maxn][20];
ll tag[maxn];
int n,k,cnt,cot;


char s[20];
int z[20];



void inserts(int len)
{
    int u=0;
    for(int j=1;j<=10-len;j++)
    {

        int c=0;
        if(!a[u][c])a[u][c]=++cnt;
        u=a[u][c];
    }
    for(int j=0;j<len;j++)
    {
        int c=s[j]-'0';
        if(!a[u][c])a[u][c]=++cnt;
        u=a[u][c];
    }
}

ll finds(int len)
{
    int u=0;

    int tt=0;
    for(int i=1;i<=10-len;i++)
    {
        int c=k;
        for(int j=1;j<=k;j++)
        {
            c=(c-1+k)%k;
            if(a[u][c])
            {
                u=a[u][c];

                break;
            }
        }
        //printf("c %d\n",c);
        z[++tt]=(c)%k;
    }
    for(int i=0;i<len;i++)
    {
        int c=k-(s[i]-'0');
        for(int j=1;j<=k;j++)
        {
            c=(c-1+k)%k;
            if(a[u][c])
            {
                u=a[u][c];

                break;
            }
        }
        //printf("c %d %d\n",c,(c+s[i]-'0')%k);
        z[++tt]=(c+s[i]-'0')%k;
    }
    int flag=0;
    for(int i=1;i<=tt;i++)
    {
        if(i==tt&&flag==0&&z[i]==0)printf("0");
        if(flag==0&&z[i]==0)continue;
        if(z[i]!=0)
        {
            flag=1;
            printf("%d",z[i]);
        }else if(flag==1)
        {
            printf("%d",z[i]);
        }

    }
    printf("\n");
}



int main()
{

    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        int len=strlen(s);
        inserts(len);
    }
    int q;
    scanf("%d",&q);
    while(q--)
    {
        scanf("%s",s);
        int len=strlen(s);
        finds(len);

    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值