洛谷 P1120 小木棍 [数据加强版]

题目来源:https://www.luogu.org/problem/show?pid=1120


在各网站提交了不下20次,终于AC了.......


一开始是用贪心做的,并没有考虑已拼接成的木棍在回溯时还会再拆开重组。程序跑得飞快但总WA几个点。在各网站提交都是WA。


后来看了题解,做法都和我不一样,但我却找不出错误。


生成了几组数据用题解对拍,才找出错误......


但会正解后,又TLE了几次,最后在题解的指引下剪了几个枝才AC。


先排序,再枚举长度,木棍从大到小依次验证是否可拼接成所枚举的长度,一旦不可行则直接回溯。


几个剪枝:


1.如果某个长度不可行,则与之相同长度的必然不行,因而不用去搜。


2.一旦拼接成功,直接输出结果,退出程序。


3.枚举木棍长度时一定要保证木棍总长能够被其整除,否则不必验证。


不同网站对>50的长度要求不同,洛谷上要求滤掉>50的长度。


代码:


#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
int n;
int a[100];
int cnt=0;
bool vis[100];
bool dfs(int num,int len,int s,int pos)
{
    int ok=1;
    if(num==1){printf("%d",s);exit(0);}
    int p=-1;
    for(int i=pos;i>=1;i--)
    {
        if(!vis[i]&&a[i]!=p)
        {
            vis[i]=1;
            if(len+a[i]<s)
            {
                if(dfs(num,len+a[i],s,i-1)){printf("%d",s);exit(0);}
                if(len==0){vis[i]=0;return 0;}               /
            }
            if(len+a[i]==s)
            {
                if(dfs(num-1,0,s,cnt)){printf("%d",s);exit(0);}
                else{vis[i]=0;break;}
            }
            vis[i]=0;
            p=a[i];
        }
    }
    return 0;
}
int main()
{
    scanf("%d",&n);
    int l=0;
    int sum=0;
    for(int i=1;i<=n;i++)
    {
        int c;scanf("%d",&c);
        if(c<=50)
        {
            sum+=c;
            a[++cnt]=c;
            l=max(l,c);
        }
    }
    sort(a+1,a+1+cnt);
    for(int i=l;;i++)
    {
        if(sum%i==0)
        {
            memset(vis,0,sizeof(vis));
            if(dfs(sum/i,0,i,cnt))
            {
                printf("%d",i);
                return 0;
            }
        }
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值