TJUACM假期集训个人赛（二）(Atcoder ABC 297A-G ABC298E-F)

这场$rk2$，考的还不错，可惜考前去上了趟厕所，前面几个题罚时炸了

A-Double Click

题面
水题，找到第一个$x_i$使得$x_i-x_{i-1}\le d$即可

#include<bits/stdc++.h>
#define reg register
#define N 300030
using namespace std;
typedef long long ll;
inline void read(int &x){
    int s=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<3)+(s<<1)+(ch&15);ch=getchar();}
    x=s*w;
}
int n,x,m,y=-1e9;
int main(){
    read(n);read(m);
    for(int i=1;i<=n;i++){
        read(x);
        if(x-y<=m){
            cout<<x<<endl;
            return 0;
        }
        y=x;
    }
    cout<<-1<<endl;
}

B-chess960

题面
水题，字符串找到两个$B$的位置，判断这两个位置是否同奇偶，不同则符合
找到$K$和两个$R$的位置 如果$K$在两个$R$中间则符合
两条都符合输出$Yes$否则输出$No$

#include<bits/stdc++.h>
#define reg register
#define N 300030
using namespace std;
typedef long long ll;
inline void read(int &x){
    int s=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<3)+(s<<1)+(ch&15);ch=getchar();}
    x=s*w;
}
char str[100];
int b,l,r,m;
int main(){
    scanf("%s",str);
    int len=strlen(str);
    for(int i=0;i<len;i++){
        if(str[i]=='B')b+=(i+1);
        if(str[i]=='R'&&l)r=i+1;
        if(str[i]=='R'&&!l)l=i+1;
        if(str[i]=='K')m=i+1;
    }
    if(b&1&&r>m&&l<m)puts("Yes");
    else puts("No");
}

C-PC on the Table

题面
这题算贪心吗，找到连着的两个$T$，把他俩变成$PC$，要求变得尽可能得多，那肯定从左往右碰到俩就变俩，这样能变出来最多的

#include<bits/stdc++.h>
#define reg register
#define N 300030
using namespace std;
typedef long long ll;
inline void read(int &x){
    int s=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<3)+(s<<1)+(ch&15);ch=getchar();}
    x=s*w;
}
char str[1000][1000];
int n,l;
int main(){
    read(n),read(l);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=l;j++){
            str[i][j]=getchar();
            while(str[i][j]!='.'&&str[i][j]!='T')str[i][j]=getchar();
            if(str[i][j]=='T'&&str[i][j-1]=='T'){
                str[i][j]='C',str[i][j-1]='P';
            }
        }
        for(int j=1;j<=l;j++)putchar(str[i][j]);
        puts("");
    }
}

D-Count Subtractions

题面
这个操作不就是辗转相减吗，给定两个整数求辗转相减的次数，那就用辗转相除，每次除法记录正常用减法一个一个减需要多少次能达到这次除的效果，边辗转相除边记录，加到一块就行

#include<bits/stdc++.h>
#define reg register
#define N 300030
using namespace std;
typedef long long ll;
inline void read(ll &x){
    ll s=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<3)+(s<<1)+(ch&15);ch=getchar();}
    x=s*w;
}
ll n,m;
ll gcd(ll a, ll b){
    if(a==b)return 0;
    if(!b)return -1;
    return a/b+gcd(b,a%b);
}
int main(){
    read(n),read(m);
    cout<<gcd(n,m)<<endl;
}

E-Kth Takoyaki Set

题面
这题直接$stl$了开一个小根堆一个$set$，先把$0$加入堆和集合，然后每次操作取堆顶，将其加上$a_1,a_2,...,a_n$，并分别将这$n$个值判断，如果在集合中没出现过则加入集合和堆，如果出现过就不加了，最后操作$m$次取堆顶即为第$m$大的。

#include<bits/stdc++.h>
#define reg register
#define N 300
using namespace std;
typedef long long ll;
inline void read(ll &x){
    ll s=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<3)+(s<<1)+(ch&15);ch=getchar();}
    x=s*w;
}
ll now,n,m,a[N];
priority_queue<ll, vector<ll>, greater<ll> > q;
set<ll> s;
int main(){
    read(n),read(m);
    for(int i=1;i<=n;i++)read(a[i]);
    q.push(0);s.insert(0);
    for(int j=1;j<=m;j++){
        now=q.top();q.pop();
        for(int i=1;i<=n;i++){
            if(s.find(now+a[i])==s.end())q.push(now+a[i]),s.insert(now+a[i]);
        }
    }
    cout<<q.top()<<endl;
}

F-Minimum Bounding Box 2

题面
一道中等难度数学题，题目大意是在一个$n\times m$的方格中随便挑选$k$个小矩形，用一个大矩形刚好覆盖这些小矩形，所得分数为这个大矩形的大小，问所有情况得分的总期望为多少。
我们不妨把问题换过来想，对于一个大矩形，它能对答案有多少贡献，答案是$f_{i,j}\ast i\ast j\ast p_{i,j}$，其中$f_{i,j}$代表矩形中包围$k$个小矩形的方案数,$p_{i,j}$代表该大矩形出现的概率。
我们利用容斥原理，现在大矩形里随便取$k$个，并排除不可行的方案，即未完全包围的方案，大矩形中第一列第一行最后一列最后一行中有空的，那么代表该情况不可行，设$a_1,a_2,a_3,a_4$为缺第一行、第一列、最后一行、最后一列的情况，$a_{12}$为缺第一行和第一列，以此类推，

$f(i,j)=C_{ij}^k-a_1-a_2-a_3-a_4+a_{12}+a_{13}+a_{14}+a_{23}+a_{24}+a_{34}-a_{123}-a_{234}-a_{341}-a_{412}+a_{1234}$

$=C_{ij}^k-2C_{(i-1)j}^k-2C_{i(j-1)}^k+C_{(i-2)j}^k+C_{i(j-2)}^k+4C_{(i-1)(j-1)}^k-2C_{(i-1)(j-2)}^k-2C_{(i-2)(j-1)}^k+C_{(i-1)(j-1)}^k$

相对来说$p(i,j)$就更好算了，所有情况为$C_{mn}^k$，矩形出现的次数为$(n-i+1)(m-j+1)$，因此$p(i,j)=\frac{(n-i+1)(m-j+1)}{C_{mn}^k}$
所以答案为$ans={\sum }_{i=1}^n{\sum }_{j=1}^{m}f(i,j)p(i,j)ij=\frac{{\sum }_{i=1}^n{\sum }_{j=1}^{m}ij(n-i+1)(m-j+1)f(i,j)}{C_{nm}^k}$

#include<bits/stdc++.h>
#define reg register
#define N 1000003
using namespace std;
typedef long long ll;
const ll mod=998244353;
inline void read(ll &x){
    ll s=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<3)+(s<<1)+(ch&15);ch=getchar();}
    x=s*w;
}
ll qpow(ll x, ll y){
    ll re=1;
    while(y){
        if(y&1)(re*=x)%=mod;
        (x*=x)%=mod,y>>=1;
    }
    return re;
}
ll frac[N],inv[N];
ll c(int n, int m){
    if(n<m)return 0LL;
    return frac[n]*inv[n-m]%mod*inv[m]%mod;
}
ll n,m,k,now,ans;
int main(){
    frac[0]=frac[1]=inv[0]=1LL;
    cin>>n>>m>>k;
    for(int i=1;i<=n*m;i++)frac[i]=frac[i-1]*i%mod,inv[i]=inv[i-1]*qpow(i,mod-2)%mod;
    for(ll i=1;i<=n;i++){
        for(ll j=1;j<=m;j++){
            now=0LL;
            for(ll l=0;l<=2;l++){
                for(ll r=0;r<=2;r++){
                    ll s=1LL;
                    if(r==1)s*=-2;
                    if(l==1)s*=-2;
                    if(i<=l||j<=r)continue;
                    (now+=(s*c((i-l)*(j-r),k)+mod))%=mod;
                }
            }
            if(now<0)now+=(-now/mod+1)*mod;
            (ans+=(now*i%mod*j%mod*(n-i+1)%mod*(m-j+1)%mod))%=mod;
        }
    }
    (ans*=qpow(c(n*m,k),mod-2))%=mod;
    if(ans<0)ans+=(-ans/mod+1)*mod;
    cout<<ans<<endl;
}

G-Constrained Nim 2

题面
博弈论，大体策略是对方拿$x$，我方拿$l+r-x$是最优解，因此$sg$函数为$sg(x)=\lfloor \frac{x\mathrm{mod}(l+r)}{l}\rfloor$，由$sg$定理，$ans=sg(a_1)\oplus sg(a_2)\oplus ...\oplus sg(a_n)$

#include<bits/stdc++.h>
#define reg register
#define N 200003
using namespace std;
inline void read(int &x){
    int s=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<3)+(s<<1)+(ch&15);ch=getchar();}
    x=s*w;
}
int l,r,x,n,ans;
int sg(int x){
    return (x%(l+r)/l);
}
int main(){
    read(n),read(l),read(r);
    for(int i=1;i<=n;i++){
        read(x);
        ans^=sg(x);
    }
    if(ans)puts("First");
    else puts("Second");
}

E - Unfair Sugoroku

题面
题意是在一个一维数轴上，$A,B$分别在$a,b$点，二者分别有个骰子，$A$的能扔出来$1-p$，$B$的能扔出来$1-q$，谁先到达大于等于$n$的点获胜，问$A$获胜的概率。

设$dp[i][j][0/1]$表示现在$A$在$i$，$B$在$j$，下一步该$A/B$走，$A$获胜的概率。则答案为$dp[a][b][0]$。
其中$i,j\le n$，对于$i,j>n$贡献可看作$i,j=n$
初始值$dp[n][i][0/1]=1,dp[i][n][0/1]=0$}
不难推出状态转移方程为
d p [ i ] [ j ] [ 0 ] = ∑ k = 1 k = p d p [ m i n { i + k , n } ] [ j ] [ 1 ] p dp[i][j][0]=\frac{\sum_{k=1}^{k=p}dp[min\{i+k,n\}][j][1]}{p} dp[i][j][0]=p∑k=1k=p​dp[min{i+k,n}][j][1]​， d p [ i ] [ j ] [ 1 ] = ∑ k = 1 k = q d p [ i ] [ m i n { j + k , n } ] [ 0 ] q dp[i][j][1]=\frac{\sum_{k=1}^{k=q}dp[i][min\{j+k,n\}][0]}{q} dp[i][j][1]=q∑k=1k=q​dp[i][min{j+k,n}][0]​

#include <bits/stdc++.h>
#define N 110
using namespace std;
typedef long long ll;
const ll mod=998244353;
inline void read(ll &x){
    ll s=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<3)+(s<<1)+(ch&15);ch=getchar();}
    x=s*w;
}
ll n,m,p,q,a,b,dp[N][N][2],ans,invp,invq;
ll qpow(ll x, ll y){
    ll re=1;
    while(y){
        if(y&1)(re*=x)%=mod;
        (x*=x)%=mod,y>>=1;
    }
    return re;
}
int main(){
    read(n);read(a);read(b);
    read(p);read(q);
    invp=qpow(p,mod-2),invq=qpow(q,mod-2);
    for(int i=1;i<=n;i++){
        for(int j=0;j<=1;j++)dp[n][i][j]=1LL,dp[i][n][j]=0LL;
    }
    for(ll i=n-1;i>=1;i--){
        for(ll j=n-1;j>=1;j--){
            for(ll k=1;k<=p;k++)(dp[i][j][0]+=dp[min(n,i+k)][j][1])%=mod;
            for(ll k=1;k<=q;k++)(dp[i][j][1]+=dp[i][min(n,j+k)][0])%=mod;
            (dp[i][j][0]*=invp)%=mod,(dp[i][j][1]*=invq)%=mod;
        }
    }
    cout<<dp[a][b][0]<<endl;
}

F - Rook Score

题面
题意是给定一个$10^9\times 10^9$的网格，在$r_i$行$c_i$列有数$x_i$，问任取一行和一列，上面的数的和最大是多少。

对于第$r$行第$c$列，我们将答案分为两种情况，第$r$行第$c$列的位置有数，和第$r$行第$c$列的位置没数。
我们先用$map$记录有数的每一行/列上所有数的和，记作$sr,sc$，
对于第一种情况直接遍历所有的数，更新答案 a n s = m a x { a n s , s r [ r i ] + s r [ c i ] − n u m [ r i ] [ c i ] } ans=max\{ans,sr[r_i]+sr[c_i]-num[r_i][c_i]\} ans=max{ans,sr[ri​]+sr[ci​]−num[ri​][ci​]}。
对于第二种情况，我们把$sr,sc$从大到小排序，每次枚举$r_i$，选出第一个$r_i$行$c_j$列没有数的$c_j$，并更新答案 a n s = m a x { s r [ r i ] + s c [ c j ] , a n s } ans=max\{sr[r_i]+sc[c_j],ans\} ans=max{sr[ri​]+sc[cj​],ans}，找到之后跳出枚举$c_j$的循环，枚举下一个$r_{i+1}$。
因为只有$n$个数，因此循环次数不会超过$2n$次，时间复杂度为$O(n)$。

#include <bits/stdc++.h>
#define N 200200
using namespace std;
typedef long long ll;
const ll mod=998244353;
inline void read(ll &x){
    ll s=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<3)+(s<<1)+(ch&15);ch=getchar();}
    x=s*w;
}

ll n,r[N],c[N],x[N],ans,cntr,cntc;
map<ll,ll> mpr;
map<ll,ll> mpc;
map<ll,set<ll> > mp;
struct node{
    ll id,sum;
}rr[N],cc[N];
bool cmp(node a, node b){
    return a.sum>b.sum;
}
int main(){
    read(n);
    for(int i=1;i<=n;i++){
        read(r[i]),read(c[i]),read(x[i]);
        mp[r[i]].insert(c[i]);
        mpr[r[i]]+=x[i],mpc[c[i]]+=x[i];
    }
    for(int i=1;i<=n;i++)ans=max(ans,mpr[r[i]]-x[i]+mpc[c[i]]);
    for(auto i:mpr)rr[++cntr].id=i.first,rr[cntr].sum=i.second;
    for(auto i:mpc)cc[++cntc].id=i.first,cc[cntc].sum=i.second;
    sort(rr+1,rr+1+cntr,cmp);sort(cc+1,cc+1+cntc,cmp);
    for(int i=1;i<=cntr;i++){
        for(int j=1;j<=cntc;j++){
            if(mp.find(rr[i].id)!=mp.end()&&mp[rr[i].id].find(cc[j].id)!=mp[rr[i].id].end())continue;
            ans=max(ans,rr[i].sum+cc[j].sum);
            break;
        }
    }
    cout<<ans<<endl;
}