Codeforces Round #205 (Div. 2) CodeForces 353B Two Heaps

CodeForces 353B

這題題意多次理解錯誤，卡了很長時間，最後看別人解析才意識到，另外codeforces上的題解沒懂什麼意思，但不妨礙做這道題。

題意：給你兩堆數，輸入初始狀態前n個數爲1堆後n個爲2堆，題目要求將這些數交換堆使得產生不同的四位數數最多，每個堆裏可以取個數和另一個堆的一個數湊在一起表示四位數。

思路：理解題意後，其實這個不同數的個數相對好求，就是輸出可能比較坑，考慮到相同的數的個數有奇數個有偶數個有一個，這樣就得分情況考慮，像偶數這個就簡單直接每堆放一半，奇數就要特判。

代碼如下：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
#define INF 0x3f3f3f3f
int num[205],b[205],a[205],v[205],s[205],x[205];
int main()
{
  int n;
  scanf("%d",&n);
  memset(num,0,sizeof(num));
  for(int i=0;i<2*n;i++)
  {
    scanf("%d",&a[i]);
    num[a[i]]++;
  }
  b[1]=b[0]=0;
  int p=1;
  for(int i=10;i<100;i++)
  {
    if(num[i]==0) continue;
    if(num[i]==1) b[p]++,p^=1;
    else b[1]++,b[0]++;
  }
  int ans=b[1]*b[0];
  memset(s,0,sizeof(s));
  memset(v,0,sizeof(v));
  memset(x,0,sizeof(x));
  p=0;int q=1;
  printf("%d\n",ans);int i;
  for(i=0;i<(2*n)-1;i++)
  {
    if(num[a[i]]==1) printf("%d ",p+1),p^=1;
    else if(num[a[i]]&1)
    {
      if(!v[a[i]])
      {
        v[a[i]]=1;s[a[i]]=q;q^=1;
      }
      else s[a[i]]^=1;
      printf("%d ",s[a[i]]+1);
    }
    else
    {
      printf("%d ",x[a[i]]+1);
      x[a[i]]^=1;
    }
  }
  if(num[a[i]]==1) printf("%d ",p+1),p^=1;
    else if(num[a[i]]&1)
    {
      if(!v[a[i]])
      {
        v[a[i]]=1;s[a[i]]=q;q^=1;
      }
      else s[a[i]]^=1;
      printf("%d ",s[a[i]]+1);
    }
    else
    {
      printf("%d ",x[a[i]]+1);
      x[a[i]]^=1;
    }
}