2020 杭电多校5 1003、1005

1003 Borrow

题意：

给三个数字 $x，y，z$，每次取最大的数字将其减一，并且等概率给剩下两个数字之一加上一，问3个数字相等的期望次数，若不可能输出 $-1$

$（\sum x+y+z\le 3\times 10^6）$

思路：

首先不整除就输出 $-1$ ，考虑$(x+y+z)/3=ave$，且 $x<=y<=z$

若$x<ave<=y<=z$ ，那么当 $x=ave$ 时，三个数字就会相等，即答案就是 $x$ 变为 $ave$ 的期望次数，设 $F(x)$ 为最小数为 $x$ 时期望的次数，那么 $F(x)=\frac{1}{2}[F(x+1)+1]+\frac{1}{2}[F(x)+1]=\frac{1}{2}F(x+1)+\frac{1}{2}F(x)+1$ ，又 $F(ave)=0$ ,通过累加可以得到， $F(x)=2\ast (ave-x)$ 。

若 $x<=y<ave<=z:$ 那么进行 $i$ 次操作， $x$ 恰好等于 $ave$ ，此时又变为上面的情况，可以根据 $F$ 求解。下面计算进行 $i$ 次操作， $x$ 等于 $ave$ 的情况。假设 $a=ave-x$ ，$b=ave-y$ ，那么发生的概率为$\frac{C_{i-1}^{a-1}}{2^i}$ (因为第i次必然是加在x，前 $i-1$ 次加了 $a-1$ 次)，在该情况下需要进行的次数就是 $i+F(y+i-a)=i+2\ast (a+b-i)$ ，那么我们只要枚举 $i$ 即可

代码：

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e6+10;
const int mod=998244353;
ll ksm(ll a,ll b) {
    ll res=1,t=a;
    while(b) {
        if(b&1) res=(res*t)%mod;
        t=(t*t)%mod;
        b>>=1;
    }
    return res;
}
ll fz[N],fm[N],p[N],np[N];
void init() {
    p[0]=1;fz[0]=1;fm[0]=1,np[0]=1;
    for(int i=1;i<N;i++) {
        p[i]=(p[i-1]*2)%mod,fz[i]=(fz[i-1]*i)%mod;
        fm[i]=ksm(fz[i],mod-2);np[i]=ksm(p[i],mod-2);
    }
}
ll f(ll x,ll k) {
    return 2*(k-x)%mod;
}
ll C(int a,int b) {
	if(a>b) return 0;
    return fz[b]*fm[a]%mod*fm[b-a]%mod;
}
int T;
ll a,b,c;

int main()
{
    init();
    scanf("%d",&T);
    while(T--) {
        scanf("%lld%lld%lld",&a,&b,&c);
        if(a>b) swap(a,b);if(b>c) swap(b,c);
        if(a>b) swap(a,b);
        if((a+b+c)%3!=0) {
            puts("-1");continue;
        }
        ll ave=(a+b+c)/3;
        if(ave<=b) {
            printf("%lld\n",f(a,ave));
        }
        else {
            ll x=ave-a,y=ave-b;
            ll ans=0;
            for(int i=1;i<x+y;i++) {
                ans=(ans+C(x-1,i-1)*np[i]%mod*(i+2*(x+y-i))%mod)%mod;
                ans=(ans+C(y-1,i-1)*np[i]%mod*(i+2*(x+y-i))%mod)%mod;
            }
            printf("%lld\n",ans);
        }
    }
}

1005 Fragrant numbers

题意：

给一个无限循环的数字串，循环串为 $"1145141919"$ ，给一个 $x$ ，问最短需要几位数字串可以构造出来，可以在数字串中插入括号，加号，乘号进行计算。例如 $x=520=1+1+4+514$

$(x<=5000)$

思路：

由于 $x<=5000$ ，猜想只需要很短的数字串就可以拼出所有的数字，可以枚举长度，对于每种长度得到其可以算出的所有数字，那么就得到了所有的答案。打表即可。

考虑 $dp[i][j]$ ，表示从第 $i$ 位到第 $j$ 位可以拼出的所有数字。使用 $set$ 记录即可。

主要对于 $dp[i][j]$ 中还需要单独加入从 $i$ 到 $j$ 位拼接成的数字。

转移就直接加和乘。

打表代码：

#include<bits/stdc++.h>
using namespace std;
const int N=5e3+10;

int ans[N];
set<int>dp[20][20];
int a[30];
int get(int x,int y) {
    int t=0;
    for(int i=x;i<=y;i++) t=t*10+a[i];
    return t;
}
void cl(int len) {
    //if(len==10)cout<<1<<endl;
    for(int i=1;i<=len;i++) for(int j=1;j<=len;j++) dp[i][j].clear();
    //if(len==10)cout<<1<<endl;
    for(int i=1;i<=len;i++) dp[i][i].insert(a[i]);

    for(int l=2;l<=len;l++) {
        for(int i=1;i<=len;i++) {
            int j=l+i-1;
            if(j>len) break;

            for(int k=i;k<j;k++) {
                for(int u:dp[i][k]) {
                    for(int v:dp[k+1][j]) {
                        if(u+v<N)dp[i][j].insert(u+v);if(v*u<N)dp[i][j].insert(v*u);
                    }
                }
            }
            if(l<=4)dp[i][j].insert(get(i,j));
        }
    }
    //if(len==10)cout<<1<<endl;
    for(int v:dp[1][len]) {
            if(v<N&&ans[v]==-1) ans[v]=len;
            //if(len==10)cout<<1<<endl;
    }
    return ;
}
int main()
{
    freopen("out.txt","w",stdout);
    memset(ans,-1,sizeof(ans));
    a[1]=1;a[2]=1;a[3]=4;a[4]=5;a[5]=1;a[6]=4;a[7]=1;a[8]=9;a[9]=1;a[10]=9;
    for(int i=11;i<30;i++) a[i]=a[i-10];
    for(int len=1;len<=14;len++) {
        cl(len);
    }
    cout<<"{";
    for(int i=1;i<=5000;i++) {
        printf("%d%c",ans[i],i==5000?'}':',');
    }
    cout<<";";
}