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);

    }
}