校赛热身 Problem C. Sometimes Naive (状压dp)

时间:2022-07-26 22:58:56

校赛热身 Problem C. Sometimes Naive (状压dp)

  1. 题解:
  2. 列举每一种3的倍数的组合,开始先求出3条边的可行解,则
  3. 六条边的可行解可以由两个三条边得来。
  4. 详见代码解析
#include<bits/stdc++.h>
using namespace std;
int a[],n,cnt,ant,is[],dp[];
struct node
{
int num,id;
}angle[];
bool cmp(const node& a,const node& b)
{
return a.num<b.num;
}
int abs1(int x)
{
if(x<)return -x;
return x;
}
void init()
{
for(int i=;i<(<<n);i++)
{
int ans=,t=i;
while(t)
{
if(t&)ans++; //这一位是1节++
t=t>>; //右移
}
if(ans%==)
{
angle[cnt].id=i;
angle[cnt++].num=ans/;
}
if(ans==)
{
int v[],ans1=,ans2=,t=i;
while(t)
{ //把二进制位是1的边对应上
if(t&)v[ans1++]=a[ans2];
t=t>>;
ans2++;
}
if(abs1(v[]-v[])<v[]&&abs1(v[]-v[])<v[]&&abs1(v[]-v[])<v[])
{ //两边只差小于第三边,和会超int
is[ant++]=i;
dp[i]=;
}
}
}
}
int main()
{
int T,cas=;scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
for(int i=;i<n;i++)
scanf("%d",&a[i]);
cnt=,ant=;
memset(dp,,sizeof(dp));
dp[]=;
init();
sort(angle,angle+cnt,cmp);//要排序
for(int i=;i<cnt;i++)
{
node e=angle[i];
for(int j=;j<ant;j++)
{
//&位运算
if(dp[e.id]&&!(e.id&is[j]))//若e.id已存在并且
//.e.id中的边与is[j]中没有重边则dp[e.id|is[j]]=1
dp[e.id|is[j]]=;//推到下一个
}
}
for(int i=cnt-;i>=;i--)
{
if(dp[angle[i].id]) //答案要最多,从后往前遍历
{
printf("Case #%d: %d\n",++cas,angle[i].num);
break;
}
}
}
return ;
}