Educational Codeforces Round 113 (Rated for Div. 2)--B. Chess Tournament

思路

• 只需求出主对角线上半部分，或者下半部分的情况就行，因为另一部分肯定与对称的部分相反
• 用一个二维数组b记录一下情况，和一个一维数组vis表示第i个人是否有赢过
• 判断b[i][j]的值分情况讨论，若a[i]==1或者a[j]==1，则b[i][j]为平局；若a[i]=2且i未胜过一局，则b[i][j]为i胜，若a[j]=2且j未胜过一局，则b[i][j]为i败
• 最后判断一下是否所有为2的人都赢过至少一局，假如否则输出NO，是则输出YES

代码

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define mod 1000000007
int vis[55],b[55][55];
char a[55];
int main(){
    int t;cin>>t;
    while(t--){
        memset(vis,0,sizeof vis);
        int n;cin>>n;
        for(int i=1;i<=n;i++)cin>>a[i];
        for(int i=1;i<=n;i++){
            for(int j=i+1;j<=n;j++){
                if(a[i]=='1'||a[j]=='1')b[i][j]=2;
                else if(a[i]=='2'&&!vis[i]) b[i][j]=1,vis[i]=1;
                else if(a[j]=='2'&&!vis[j]) b[i][j]=0,vis[j]=1;
                else b[i][j]=1;
            }
        }
        int flag=1;
        for(int i=1;i<=n;i++){
            if(!vis[i]&&a[i]=='2'){
                flag=0;break;
            }
        }
        if(!flag){
            cout<<"NO"<<endl;
            continue;
        }
        cout<<"YES"<<endl;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(i==j)cout<<'X';
                else if(i>j){
                    if(b[j][i]==1)cout<<'-';
                    else if(b[j][i]==2)cout<<'=';
                    else cout<<'+';
                }
                else if(i<j){
                    if(b[i][j]==0)cout<<'-';
                    else if(b[i][j]==2)cout<<'=';
                    else cout<<'+';
                }
            }
            cout<<endl;
        }
    }
}