Tinkoff Challenge – Final Round (Codeforces Round #414, rated, Div. 1 + Div. 2), problem: (G) Replace All Solution In C/C++

#include<set>
#include<map>
#include<ctime>
#include<cmath>
#include<queue>
#include<bitset>
#include<string>
#include<cstdio>
#include<cctype>
#include<vector>
#include<cstdlib>
#include<cstring>
#include<cassert>
#include<sstream>
#include<numeric>
#include<iostream>
#include<algorithm>
#include<functional>
using namespace std;
#define For(i,x,y) for(int i=x,_y=y;i<=_y;++i)
#define Rep(i,x,y) for(int i=x,_y=y;i>=_y;–i)
#define pub push_back
#define pob pop_back
#define fi first
#define se second
#define mp make_pair
typedef long long ll;
typedef double db;
typedef pair<int,int> pii;
typedef vector<int> vi;
inline int read()
{
int x=0;char c=getchar();bool f=0;
for(;c>’9’||c<‘0′;f=c==’-‘,c=getchar());
for(;c>=’0’&&c<=’9’;x=(x<<1)+(x<<3)+c-‘0’,c=getchar());
return f?-x:x;
}
inline void write(int x)
{
if(x<0)putchar(‘-‘),x=-x;
if(x>=10)write(x/10);
putchar(x%10+48);
}
const int mo=1e9+7,N=1e6+5;
int gcd(int a,int b)
{
return b?gcd(b,a%b):a;
}
void renew(int &x,int y)
{
if((x+=y)>=mo)x-=mo;
}
int Pow(int a,int b)
{
int res=1;
for(;b;b>>=1,a=(ll)a*a%mo)if(b&1)res=(ll)res*a%mo;
return res;
}
int fac[2*N],inv[2*N],pw[2*N],p[2*N],g[2*N],phi[2*N];
bool vis[2*N];
void prepare(int n)
{
fac[0]=inv[0]=pw[0]=1;
For(i,1,n)fac[i]=(ll)fac[i-1]*i%mo;
inv[n]=Pow(fac[n],mo-2);
Rep(i,n-1,1)inv[i]=(ll)inv[i+1]*(i+1)%mo;
For(i,1,n)pw[i]=pw[i-1]*2%mo;
g[1]=phi[1]=1;vis[1]=1;
For(i,2,n)
{
if(!vis[i])p[++*p]=i,phi[i]=i-1;
for(int j=1;j<=*p&&(ll)i*p[j]<=n;++j)
{
vis[i*p[j]]=1;
if(i%p[j])phi[i*p[j]]=phi[i]*(p[j]-1);else
{
phi[i*p[j]]=phi[i]*p[j];
break;
}
}
}
For(i,2,n)g[i]=(g[i-1]+phi[i]*2)%mo;
}
int n,m,la,lb,ans;
char a[N],b[N];
int C(int n,int m)
{
return (ll)fac[n]*inv[m]%mo*inv[n-m]%mo;
}
int calc1(int c0,int c1)
{
if(c0<0)c0*=-1,c1*=-1;
int d=gcd(c0,c1);
c0/=d,c1/=d;
return (pw[n/max(c0,c1)+1]-2+mo)%mo;
}
int calc2(int c)
{
int mi=1,res=0;
For(i,1,la)
{
if(a[i]==’?’&&b[i]==’?’)mi=mi*2%mo;else
if(a[i]!=’?’&&b[i]!=’?’&&a[i]!=b[i]){mi=0;break;}
}
renew(c,mo-mi);
renew(res,(ll)(pw[n+1]-2+mo)*(pw[n+1]-2+mo)%mo*mi%mo);
For(i,1,n)renew(res,(ll)g[n/i]*c%mo*pw[i]%mo);
return res;
}
int a0,a1,a2,b0,b1,b2;
int main()
{
prepare(2000000);
scanf(“%s%s%d”,a+1,b+1,&n);
la=strlen(a+1);lb=strlen(b+1);
For(i,1,la)
{
if(a[i]==’A’)++a0;else
if(a[i]==’B’)++a1;else
if(a[i]==’?’)++a2;
}
For(i,1,lb)
{
if(b[i]==’A’)++b0;else
if(b[i]==’B’)++b1;else
if(b[i]==’?’)++b2;
}
int cnt=0;
For(i,-b2,a2)
{
int t1=a0-b0+i,t2=b1-a1+b2-a2+i,c=C(a2+b2,i+b2);
if(!t1&&!t2)
{
renew(ans,calc2(c));
assert(++cnt<2);
}else
if((ll)t1*t2>0)
{
renew(ans,(ll)c*calc1(t1,t2)%mo);
}
}
printf(“%d\n”,ans);
}

As the Tide Turns...

The ongoing conflict in Ukraine has shifted from a...

KOOPAO 69W 4-in-1 Retractable...

In today's fast-paced world, keeping your mobile devices fully...

Travis Kelce’s Stylist Arrives...

The rumor mill is spinning faster than ever for...

US plans return to...

In a historic move that could reshape global aviation,...

Drafting Illinois’ biggest rivals!

Who is the Ultimate Rival of the Illinois Fighting...

4DMedical Receives TGA Approval...

4DMedical Limited (ASX: 4DX), a global leader in advanced...

As the Tide Turns Against Putin, Beware the Drowning Man

The ongoing conflict in Ukraine has shifted from a predicted swift victory into a grueling war of attrition, creating an increasingly precarious situation for...

KOOPAO 69W 4-in-1 Retractable Car Charger w/ USB-C & Lightning Cables $6.50

In today's fast-paced world, keeping your mobile devices fully charged while on the road is more of a necessity than a luxury. However, dealing...

Travis Kelce’s Stylist Arrives in NYC Ahead of Reported Taylor Swift Wedding

The rumor mill is spinning faster than ever for Kansas City Chiefs tight end Travis Kelce and pop superstar Taylor Swift. Speculation of an...

US plans return to supersonic flights with new FAA rule; to reverse 53-year-old ban

In a historic move that could reshape global aviation, the United States is paving the way for the return of commercial supersonic flight over...

Drafting Illinois’ biggest rivals!

Who is the Ultimate Rival of the Illinois Fighting Illini? For decades, sports fans have debated the true identity of the Illinois Fighting Illini's ultimate...

4DMedical Receives TGA Approval for CT:VQ™, Bringing Non-Contrast Ventilation-Perfusion Imaging to Australia

4DMedical Limited (ASX: 4DX), a global leader in advanced respiratory imaging software, has officially received regulatory approval from the Australian Therapeutic Goods Administration (TGA)...

Raw Milk Sickened More Than 80 People in Six Months: Here’s What Pasteurization Actually Prevents

The Risks of Raw Milk: Examining the 2026 Outbreaks In recent years, the raw milk movement has gained significant traction among health enthusiasts seeking "unprocessed"...

Baylor College of Medicine Scientists Find Tubulin Protein Prevents Tau and Alpha-Synuclein Clumping in Alzheimer’s and Parkinson’s

In a groundbreaking development for neurodegenerative disease research, scientists at the Baylor College of Medicine have uncovered a critical mechanism that could redefine how...

THN Archive: A Flip Of The Subban Switch

In the world of professional hockey, few players have managed to capture the hearts of fans both on and off the ice quite like...