题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5355
题面:
4 1 2 5 3 5 2 9 3
NO YES 1 5 2 1 4 2 2 3 NO YES 3 1 5 9 3 2 6 7 3 3 4 8
#include <iostream>
#include <cstdio>
#include <cstring>
#define LL long long
#define mod 1000000007
using namespace std;
//访问和成功标记
bool status[45],sign;
//记录搜索路径
int path[45];
//平均值,cnt为当前第几组解
int ave,m,n,cnt;
//flag[x][y]表示x块蛋糕能不能划分为y部分
bool flag[45][12];
struct division
{
int a[45],sz;
}store[45][12][12];
//三维分别表示共多少,划分为几块,它的第几块是,以及内部存储分别是哪几块
//预处理,搜索
//pos为当前位置,left为剩余容量,p为路径中第几个数字
void dfs(int pos,int left,int p)
{
//当前状态下已经找到一种方案
if(sign)return;
//成功找到解
if(left==0)
{
//把路径中的值存入到数组中
for(int i=0;i<p;i++)
store[n][m][cnt].a[i]=path[i];
store[n][m][cnt].sz=p;
//成功标记
sign=true;
return;
}
for(int i=pos;i>=1;i--)
{
if(!status[i]&&left>=i)
{
//该点已访问
status[i]=1;
path[p]=i;
dfs(i,left-i,p+1);
//避免把已经成功的点重新置为0
if(sign)return;
status[i]=0;
}
}
}
//预处理
void prep()
{
memset(flag,0,sizeof(flag));
int tmp,total;
for(int i=1;i<=20;i++)
{
//tmp是根据n要小于等于平均值,计算出来的上界
tmp=(i+1)/2;
total=i*(i+1)/2;
for(int j=1;j<=tmp;j++)
{
//如果可以划分的话
if(total%j==0)
{
flag[i][j]=1;
memset(status,0,sizeof(status));
ave=total/j;
m=j;
n=i;
//循环找多组解
for(int k=0;k<j;k++)
{
cnt=k;
sign=false;
dfs(n,ave,0);
}
}
}
}
}
int main()
{
//预处理
prep();
int t,x,y,a,b,sz;
//读入
scanf("%d",&t);
LL total;
while(t--)
{
scanf("%d%d",&x,&y);
total=1LL*(1+x)*x/2;
ave=total/y;
//排除不能划分的情况
if((total%y)||(ave<x))
{
printf("NO\n");
continue;
}
//其余的都是可构造的
printf("YES\n");
m=y;
//注意m已经乘以2
m<<=1;
int g;
//找到第一个合法的状态
for(g=1;;g++)
{
//如果可以划分,而且后面多出的数刚好可以构造
if(flag[g][y]&&((x-g)%m)==0)
break;
}
//b为构造次数
b=(x-g)/m;
for(int i=0;i<y;i++)
{
sz=store[g][y][i].sz;
//总数量
printf("%d",sz+(b<<1));
//原有解
for(int k=0;k<sz;k++)
printf(" %d",store[g][y][i].a[k]);
//构造解
for(int k=0;k<b;k++)
printf(" %d %d",k*m+g+i+1,(k+1)*m+g-i);
printf("\n");
}
}
return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。
HDU 5355 Cake (WA后AC代码,详细解析,构造题)
原文地址:http://blog.csdn.net/david_jett/article/details/47358057