QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#211801 | #7400. Digital Root | SolitaryDream | WA | 1ms | 13984kb | C++17 | 3.9kb | 2023-10-12 21:22:42 | 2023-10-12 21:22:43 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
#define int long long
using pii=pair<int,int>;
const int N=1e6+1e3+7;
int cnt[16][1<<16];
int n,m,B;
char S[N];
int a[N],s[N];
vector<int> pos[16];
int dp[N][2],F[2];
int g[N][2][16][2],G[2][16][2];
signed main()
{
scanf("%lld%lld%lld",&n,&m,&B);
scanf("%s",S+1);
int cz=0,sz=0,nz=0,bz=0;
int W=B-1;
for(int i=1;i<=n;i++)
{
if(S[i]=='0')
{
for(int j=0;j<2;j++)
dp[i][j]=dp[i-1][j];
}
else
{
dp[i][1]=dp[i-1][0];
}
dp[i][S[i]!='0']++;
for(int j=0;j<2;j++)
F[j]+=dp[i][j];
if(S[i]=='0')
sz++;
else
sz=0;
cz+=sz;
a[i]=islower(S[i])?S[i]-'a'+10:S[i]-'0';
bz+=a[i]==B-1;
a[i]%=(B-1);
if(S[i]=='0')
{
for(int j=0;j<2;j++)
for(int k=0;k<W;k++)
g[i][j][(k+a[i])%W][1]=g[i-1][j][k][0]+g[i-1][j][k][1];
}
else if(a[i]!=0)
{
for(int j=0;j<W;j++)
g[i][1][(j+a[i])%W][1]=g[i-1][0][j][0]+g[i-1][0][j][1];
}
if(S[i]=='0'||a[i]!='0')
g[i][a[i]!=0][a[i]][0]++;
for(int j=0;j<2;j++)
for(int k=0;k<W;k++)
for(int t=0;t<2;t++)
G[j][k][t]+=g[i][j][k][t];
nz+=S[i]!='0';
}
vector<pii>st;
pos[0].push_back(0);
for(int i=1;i<=n;i++)
{
s[i]=(s[i-1]+a[i])%W;
pos[s[i]].push_back(i);
st.push_back({1<<a[i],i});
for(auto &[x,y]:st)
x|=1<<a[i];
vector<pii> nst;
reverse(st.begin(),st.end());
for(auto [x,y]:st)
{
if(nst.empty()||x!=nst.back().first)
nst.push_back({x,y});
}
st.swap(nst);
reverse(st.begin(),st.end());
for(int y=(int)st.size()-1;y>=0;y--)
{
int R=st[y].second,L=y>0?st[y-1].second+1:1;
for(int j=0;j<W;j++)
{
int v=(s[i]-j+W)%(W);
int A=upper_bound(pos[v].begin(),pos[v].end(),R-1)-lower_bound(pos[v].begin(),pos[v].end(),L-1);
// if(j==2&&A)n
cnt[j][st[y].first]+=A;
}
}
}
for(int t=0;t<W;t++)
for(int i=0;i<W;i++)
for(int S=0;S<(1<<W);S++)
if(!(S>>i&1))
cnt[t][S^(1<<i)]+=cnt[t][S];
while(m--)
{
scanf("%s",S);
int x=islower(S[0])?S[0]-'a'+10:S[0]-'0';
int bx=x;
x%=W;
scanf("%s",S);
if(bx==0)
{
int p=strlen(S);
bool ok=0;
for(int j=0;j<p;j++)
if(S[j]=='0')
ok=1;
printf("%lld\n",F[0]+ok*(F[1]));
continue;
}
int p=strlen(S);
int E=0;
for(int i=0;i<p;i++)
E|=1<<(islower(S[i])?S[i]-'a'+10:S[i]-'0');
long long ans=0;
ans+=cnt[x][(1<<W)-1];
if(x==0&&!(E&(1<<W)))
ans-=F[0];
if(E&(1<<W))
{
E^=(1<<W);
E|=1;
}
else if((E&1)&&!(E&(1<<W)))
{
if(x==0)
{
for(int t=0;t<W;t++)
{
if(!(E>>t&1))
ans-=G[1][(W-t)%W][1];
ans-=G[1][t][0];
}
}
}
for(int j=0;j<W;j++)
{
if(j==x)
continue;
int s=(x-j+W)%(W);
int nS=E>>s|((E<<(W)>>s)&((1<<W)-1));
nS^=(1<<W)-1;
ans+=cnt[j][(1<<W)-1]-cnt[j][nS];
}
printf("%lld\n",ans);
}
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 13984kb
input:
9 2 10 123456789 9 12 8 123456789
output:
24 45
result:
ok 2 tokens
Test #2:
score: 0
Accepted
time: 1ms
memory: 11984kb
input:
5 10 5 01234 0 1 1 1 2 1 3 1 4 1 0 1 1 0 2 0 3 0 4 0
output:
1 13 9 9 9 1 10 9 10 6
result:
ok 10 tokens
Test #3:
score: 0
Accepted
time: 1ms
memory: 9820kb
input:
19 6 2 0010111001010011111 0 0 0 1 0 01 1 0 1 1 1 01
output:
42 11 42 179 190 190
result:
ok 6 tokens
Test #4:
score: 0
Accepted
time: 1ms
memory: 9760kb
input:
13 21 3 1010011001002 0 0 0 1 0 10 0 2 0 20 0 21 0 012 1 0 1 1 1 01 1 2 1 20 1 12 1 021 2 0 2 1 2 01 2 2 2 02 2 12 2 102
output:
36 10 36 10 36 10 36 78 90 91 78 78 91 91 58 76 76 91 91 91 91
result:
ok 21 tokens
Test #5:
score: -100
Wrong Answer
time: 0ms
memory: 11872kb
input:
15 60 4 313213200103021 0 0 0 1 0 01 0 2 0 20 0 21 0 021 0 3 0 30 0 31 0 103 0 23 0 032 0 321 0 0132 1 0 1 1 1 10 1 2 1 02 1 12 1 021 1 3 1 30 1 13 1 310 1 23 1 203 1 312 1 1302 2 0 2 1 2 01 2 2 2 02 2 21 2 120 2 3 2 03 2 13 2 031 2 32 2 320 2 213 2 0321 3 0 3 1 3 01 3 2 3 20 3 12 3 201 3 3 3 30 3 1...
output:
27 5 27 5 27 5 27 5 27 5 27 5 27 5 27 96 118 120 105 105 120 120 96 96 120 120 105 105 120 120 89 104 104 118 120 120 120 89 89 104 104 120 120 120 120 94 93 102 99 100 108 108 120 120 120 120 120 120 120 120
result:
wrong answer 46th words differ - expected: '100', found: '94'