学习笔记4：（AtCoder Beginner Contest 326）DEF

D - ABC Puzzle (atcoder.jp)

要求：

在一个N*N网格操作，初始全是"."

给一个长度为N的R和C

1.每行每列必须且仅包含  1个A 1个B 1个C  

2.i行最左边第一个出现的字符必须是Ri

  j列最上边第一个出现的字符必须是Cj

数据范围：3<=N<=5

分析：第一眼爆搜，但是实际实现方案数太多，所以选择预处理＋位运算优化

首先，我们预处理出所有合法字符串x,也就是包含一个A，B，C，其余是"."的字符串x,把它放入它的第一个非"."首字符对应的数组v [ x [ i ] - ' A ' ]中。

然后搜索，在搜索的过程中用到了位运算优化，1111为一个字符位，前三位判断的是 A B C，最后一位判断的是是否是该列首个非"."字符。

#include<bits/stdc++.h>
using namespace std;
#define int long long
vector<vector<string>> row(3);
bool fnd=false;
int n;
string r,c;
vector<string>v[3],ans;
int f;

void dfs(int u,int now){
    if(f) return ;
    if(u==n){
        //cout<<"hh"<<endl;
        if(now)return ;
        cout<<"Yes"<<endl;
        for(int i=0;i<n;i++){
                cout<<ans[i]<<endl;
        }
        f=1;
        return;
    }
    for(auto cc:v[r[u]-'A']){
        int getnow=now;
        ans.push_back(cc);
        int flag=0;
        for(int j=0;j<n;j++){
            char t=cc[j];
            if(t=='.') continue;
            int tmp=1<<(4*j+(t-'A'));//移到哪一位
            if(now&tmp==0){
                flag=1;
                break; 
            } 
            getnow^=tmp;
            int test_tmp=1<<(4*j+3);//测试这行是否有字符，如果有就是第一个
            if(now&test_tmp){
                if(t!=c[j]){
                    flag=1;
                    break; 
                }
                getnow^=test_tmp;
            }
        }
        if(!flag) dfs(u+1,getnow);
        ans.pop_back();
    }
}
signed main(){
    cin>>n>>r>>c;
    string x="ABC";
    while(x.size()<n)x.push_back('.');
    sort(x.begin(),x.end());
    do{
        char tmp='.';
        for(int i=0;i<n;i++){
            if(x[i]!='.'){
                tmp=x[i];
                break;
            }
        }
        v[tmp-'A'].push_back(x);
    }while(next_permutation(x.begin(),x.end()));
    dfs(0,(1<<(4*n))-1);
    if(!f)cout<<"No"<<endl;
}

E - Revenge of "The Salary of AtCoder Inc."

概率和期望不太会算qwq

两种实现方式，正序和倒序

当前的期望为上一个事件的期望之和*Pi再加上当前的贡献

倒序(意义是最终在哪个数结束，如在n结束只有一种可能  1 2 3 ……n n)

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
#define int long long 
const int N = 1000010,mod=998244353;
int a[N];
int qmi(int a,int b){
    int res=1;
    while(b){
        if(b&1) res=(res*a)%mod;
        a=(a*a)%mod;
        b>>=1;
    }
    return res;
    
}
signed main(){
    int n,m;
    cin>>n;
    
    int inv=qmi(n,mod-2);
    for(int i=1;i<=n;i++) cin>>a[i];
    int ans=0;
    for(int i=n;i>=1;i--){
        ans=(ans+ans*inv%mod+a[i])%mod;
    }
    cout<<(ans*inv)%mod<<endl;
    
}

正序（意义上结束前的那个数）

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
#define int long long 
const int N = 1000010,mod=998244353;
int a[N];
int qmi(int a,int b){
    int res=1;
    while(b){
        if(b&1) res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}
signed main(){
    int n;
    cin>>n;
    int inv=qmi(n,mod-2);
    int res=0,p=inv;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        res=(res+(p*a[i])%mod)%mod;
        p=(p+(p*inv)%mod)%mod;
    }
    cout<<res<<endl;
}

F - Robot Rotation (atcoder.jp)

非常有意思的一道题

代码实现参考：ABC326F Robot Rotation - yxcat 的垃圾堆 - 洛谷博客 (luogu.com.cn)

分析：80数据范围 爆搜2^80时间复杂度不合理

我们仔细观察会发现在奇数次操作时，只能向上或向下，在偶数则只能左右，分开爆搜2*2^40还是会超时，这是我们可以把单纯爆搜替换成双向搜索(2*2^20)*(2*2^20),这个事件复杂度是可以接受的。在实现时,f奇数次操作和偶数次操作分别进行，可以使用位运算优化，本题还有一个细节是在输出时的方向。

#include <iostream>
#include <cstring>
#include <algorithm>
#include <map>
#include <vector>

using namespace std;
#define int long long
const int N = 1000010;
int a[N],b[N];
int n,x,y;
int na=0,nb=0;
char ans[N];
map<int,int>mp;
int ansa=-1,ansb=-1;

void solve()
{
    int sum=0;
    for(int i=1;i<=na;i++) sum+=-a[i];
    int lim=na/2;
    mp.clear();
    for(int i=0;i<(1<<lim);i++)
    {
        int res=0;
        for(int j=0;j<lim;j++)
            if(i>>j&1) res+=2*a[j+1];
        mp[res]=i;
    }
    for(int i=0;i<(1<<(na-lim));i++)
    {
        int res=0;
        for(int j=0;j<na-lim;j++)
            if(i>>j&1) res+=2*a[j+lim+1];
        if(mp.count(x-(sum+res))) {ansa=(i<<lim)+mp[x-sum-res];return;}
    }
}

signed main(){
    cin>>n>>y>>x;
    for(int i=1;i<=n;i++) 
    {
        if(i&1) cin>>a[++na];
        else cin>>b[++nb];
    }
    solve();
    swap(a,b),swap(na,nb),swap(ansa,ansb),swap(x,y);
    solve();
    swap(ansa,ansb);
    int to=1;
    int aa=0,bb=0;
    if(ansa==-1||ansb==-1) {printf("No\n");return 0;}
    printf("Yes\n");

    for(int i=1;i<=n;i++){
        
         if(i&1)
        {
            int now=ansa>>aa&1;
            if(now==0) ans[i]=((to==1)?'R':'L'),to=4;
            else ans[i]=(to==1)?'L':'R',to=2;
            aa++;
        }
        else
        {
            int now=ansb>>bb&1; bb++;
            if(now==0) ans[i]=(to==2)?'L':'R',to=3;
            else ans[i]=(to==2)?'R':'L',to=1;
        }
    }
    for(int i=1;i<=n;i++)cout<<ans[i];
    
    
}