注意到前3个操作都不会影响每列的情况,而第4个操作必然会将行列交换,故只要每行的和相同即可满足条件。
考虑数位DP,设$f[i][j][k][t]$表示考虑最高的$i$位,第一行的和是$j$,当前行的和是$k$,与$R$的大小关系为$t$的数字个数。
需要特判$|R|=1$以及$|R|$不是完全平方数的情况。
#include<cstdio>
int n,m,s,a[20],i,j,k,x,t,nx;long long R,f[20][40][40][2],ans;
int main(){
scanf("%lld",&R);
if(R<10)return printf("%lld",R+1),0;
while(R)a[n++]=R%10,R/=10;
if(n!=4&&n!=9&&n!=16)return puts("0"),0;
for(m=1;m*m<n;m++);
s=m*9;
for(i=1;i<=a[n-1];i++)f[n-1][0][i][i==a[n-1]]++;
for(i=n-1;i;i--)for(j=0;j<=s;j++)for(k=0;k<=s;k++)for(x=0;x<2;x++)if(f[i][j][k][x])
for(t=0;t<10;t++){
if(x&&t>a[i-1])continue;
nx=x&&t==a[i-1];
if(i%m)f[i-1][j][k+t][nx]+=f[i][j][k][x];
else if(i/m==(n-1)/m)f[i-1][k][t][nx]+=f[i][j][k][x];
else if(j==k)f[i-1][j][t][nx]+=f[i][j][k][x];
}
for(j=0;j<=s;j++)for(x=0;x<2;x++)ans+=f[0][j][j][x];
return printf("%lld",ans),0;
}