Codeforces Round #540 (Div. 3)--1118C - Palindromic Matrix

https://codeforces.com/contest/1118/problem/C

在查找元素的时候，必须按4,2,1的顺序进行。因为，如果先找1，可能就把原来的4拆散了，然后再找4，就找不到了

#include<bits/stdc++.h>
using namespace std;
int ans[][];
int main(){
    int n;
    cin>>n;
    map<int,int> mp;
    int t;
    for(int i=;i<n*n;i++){
        cin>>t;
        mp[t]++;
    }
    vector<pair<int,pair<int,int> > > ceil;
    for(int i=;i<(n+)/;i++)
        for(int j=;j<(n+)/;j++){
            if(i!=n--i&&j!=n--j)
                ceil.push_back({,{i,j}});
            else if(i!=n--i||j!=n--j)
                ceil.push_back({,{i,j}});
            else
                ceil.push_back({,{i,j}});
        }
    int number[]={,,};
    for(int q=;q<;q++){
        int w=number[q];
        for(int v=;v<ceil.size();v++){
            pair<int,pair<int,int> > c=ceil[v];
            int times=c.first;
            if(times!=w)
                continue;
            int i=c.second.first;
            int j=c.second.second;
            map<int,int>::iterator it;
            for(it=mp.begin();it!=mp.end();it++){
                if(it->second>=times){
                    it->second-=times;
                    ans[i][j] = ans[n - i - ][j] = ans[i][n - j - ] = ans[n - i - ][n - j - ] = it->first;
                    break;
                }
            }
            if(it==mp.end()){
                cout<<"NO"<<endl;
                return ;
            }
        }
    }
    cout<<"YES"<<endl;
    for(int i=;i<n;i++){
        for(int j=;j<n;j++){
            cout<<ans[i][j]<<" ";
        }
        cout<<endl;
    }
    return ;
}