Codeforces LATOKEN Round 1 (Div. 1 + Div. 2)

06.15 1:54 比赛还没结束，先发私密，明早起来再开放
传送门

A. Colour the Flag

题解

当至少一个点已经确定，就可以直接推整张图。

代码

char g[51][51];
int n,m;
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
bool dfs(int x,int y) {
    for(int i=0;i<4;++i) {
        int xx=x+dx[i],yy=y+dy[i];
        if(xx<0||yy<0||xx>=n||yy>=m) continue;
        if(g[xx][yy]=='.') {
            if(g[x][y]=='R') g[xx][yy]='W';
            else g[xx][yy]='R';
            if(dfs(xx,yy)==false) return false;
        }
        else if(g[xx][yy]==g[x][y]) return false;
    }
    return true;
}
bool solve() {
    cin>>n>>m;
    for(int i=0;i<n;++i) cin>>g[i];
    int cnt=0;
    for(int i=0;i<n;++i) {
        for(int j=0;j<m;++j) {
            if(g[i][j]!='.') {
                return dfs(i,j);
            }
        }
    }
    g[0][0]='W';
    return dfs(0,0);
}
int main() {
    int T;
    cin>>T;
    while(T--) {
        if(solve()) {
             puts("YES");
             for(int i=0;i<n;++i) cout<<g[i]<<endl;;
        } 
        else {
            puts("NO");
        }
    }
    return 0;
}

B. Histogram Ugliness

题解

容易发现一个结论：只有一个塔比它左右都高，减它才有收益。

代码

const int N=500010;
int n,a[N];
PII b[N];
bool solve() {
    cin>>n;
    long long ans=0;
    a[0]=a[n+1]=0;
    for(int i=1;i<=n;++i) {
        cin>>a[i];
        b[i].first=a[i],b[i].second=i;
        ans=(ans+abs(a[i]-a[i-1]));
    }
    ans+=a[n];
    sort(b+1,b+n+1);
    for(int i=n;i>=1;--i) {
        int j=b[i].second;
        if(a[j-1]<a[j]&&a[j]>a[j+1]) {
            ans-=a[j]-max(a[j-1],a[j+1]);
        }
    }
    cout<<ans<<endl;
    return true;
}

C. Little Alawn’s Puzzle

题解

两个序列都是完整的排列，当交换对应位置的两个数，序列都不完整了，需要继续移动，一次移动的两个数而导致移动的一系列对可以看作一个集合，答案就是$2^{集合的个数}$，用冰茶几求集合。

代码

const int N=500010,mod=1e9+7;
int n,p[N],a[N][2],b[N][2];
int f[N];
int Find(int x) {
    return f[x]==x?x:f[x]=Find(f[x]);
}
bool solve() {
    scanf("%d",&n);
    for(int i=1;i<=n;++i) f[i]=i;
    for(int i=1;i<=n;++i) scanf("%d",&a[i][0]),b[a[i][0]][0]=i;
    for(int i=1;i<=n;++i) scanf("%d",&a[i][1]),b[a[i][1]][1]=i;
    
    for(int i=1;i<=n;++i) {
        int x=a[i][0],y=a[i][1];
        int j=b[x][1],k=b[y][0],ii=i;
        j=Find(j),k=Find(k),ii=Find(ii);
        f[j]=ii;
        f[k]=ii;
    }
    int cnt=0;
    for(int i=1;i<=n;++i) {
        if(Find(i)==i) cnt++;
    }
    cout<<p[cnt]<<endl;
    return true;
}
int main() {
    int T;
    p[0]=1;
    for(int i=1;i<N;++i) p[i]=1LL*p[i-1]*2%mod;
    scanf("%d",&T);
    while(T--) {
        if(solve()) {
            // puts("YES");
        } 
        else {
            //puts("NO");
        }
    }
    return 0;
}

D. Lost Tree

题解

随便询问一个节点，都可以将数分层，对于所有相邻的两层i,j，至少有一层被完全询问，通过线性dp求，从1层开始，每一层有两种状态查询或不查询，权值是每一层的节点数，最终满足权值<=(n+1)/2，另外dp的过程还要维护一下状态转移的过程。
dp完了之后就可以知道最后一层的查询状态，然后根据记录的转移过程到上一层继续查询或不查询。

代码

typedef pair<int,int> PII;
const int N=2010;
int f[N][2];
int g[N][2],n;
int ge[N],have=0,cnt[N];
vector<int> v[N];
set<PII> E;
void query(int t) {
    have++;
    cout<<"? "<<t<<endl;
    for(int i=1;i<=n;++i) cnt[i]=0;
    for(int i=1;i<=n;++i) {
        cin>>ge[i],cnt[ge[i]]++;
        if(ge[i]==1) {
            E.insert(make_pair(min(t,i),max(t,i)));
        }
    }
}
bool check(int root) {
    query(root);
    g[0][1]=1;
    g[0][0]=1e9;
    for(int i=1;i<=n;++i) {
        g[i][0]=g[i-1][1];
        f[i][0]=1;
        //g[i][1]
        if(g[i-1][1]<g[i-1][0])  {
            g[i][1]=g[i-1][1]+cnt[i];
            f[i][1]=1;
        }
        else {
            g[i][1]=g[i-1][0]+cnt[i];
            f[i][1]=0;
        }
    }
    return min(g[n][0],g[n][1])+have-1<=(n+1)/2;
}
int main() {
    unsigned seed; 
    seed = time(0);
    srand(seed);
    cin>>n;
    while(true) {
        int t=rand()%n+1;
        if(check(t)) break;
    }
    for(int i=1;i<=n;++i) {
        v[ge[i]].push_back(i);
    }
    int i=n,j;
    if(g[n][0]<g[n][1]) j=0;
    else j=1;
    int es=0;
    while(i>=1) {
        if(j) {
            for(int k=0;k<v[i].size();++k) {
                query(v[i][k]);
                for(int kk=1;kk<=n;++kk) {
                    if(ge[kk]==1) {
                        E.insert(make_pair(min(v[i][k],kk),max(v[i][k],kk)));
                    }
                }
            }   
        }
        j=f[i][j];
        i--;
    }
    cout<<"! "<<endl;
    set<PII>::iterator it=E.begin();
    while(it!=E.end()) {
        cout<<it->first<<' '<<it->second<<endl;
        it++;
    }
}