360笔试第二题解法

#include <bits/stdc++.h>
using namespace std;

int main()
{
   int n,m;
   cin>>n>>m;
   vector<int> A(n,0);
   vector<bool> sA(n,false);
   unordered_map<int,int> B;
   int value;

   for(int i=0;i<n;i++)
   {
       cin>>A[i];
       A[i]=A[i]%5;

   }
   for(int i=0;i<n;i++)
   {
       cin>>value;
       value=value%5;
       B[value]++;
   }

   int num=m;
   while(m--)          //从最大的数m-1开始找
   {
       if(!B.size())        //当哈希表中没有元素了，说明已经找完了
           break;

       for(int i=0;i<n;i++)
       {
           if(sA[i])
                continue;   //对已经匹配好的A中的元素，不需要在匹配了

           int temp=m;
           if(m<A[i])            //如果m的值小于A[i]，那么m-A[i]就是负数，肯定不行
               temp=m+num;

           if(B[temp-A[i]]>=1)       //与A[i]和为num的元素在哈希表B中是否存在，存在，就对和m求余存在A[i]中，然后A[i]设置为已经处理了
           {
               B[temp-A[i]]--;
               A[i]=temp%num;
               sA[i]=true;
           }
       }
   }
   sort(A.begin(),A.end(),[](int a,int b){return a>b;});
   for(auto c:A)
      cout<<c<<' ';
}