P1118 [USACO06FEB]数字三角形`Backward Digit Su`… 回溯法

有这么一个游戏：

写出一个11至NN的排列a_iai，然后每次将相邻两个数相加，构成新的序列，再对新序列进行这样的操作，显然每次构成的序列都比上一次的序列长度少11，直到只剩下一个数字位置。下面是一个例子：

3,1,2,43,1,2,4

4,3,64,3,6

7,97,9

1616

最后得到1616这样一个数字。

现在想要倒着玩这样一个游戏，如果知道NN，知道最后得到的数字的大小sumsum，请你求出最初序列a_iai，为11至NN的一个排列。若答案有多种可能，则输出字典序最小的那一个。

[color=red]管理员注：本题描述有误，这里字典序指的是1,2,3,4,5,6,7,8,9,10,11,121,2,3,4,5,6,7,8,9,10,11,12

而不是1,10,11,12,2,3,4,5,6,7,8,91,10,11,12,2,3,4,5,6,7,8,9[/color]

输入输出格式

输入格式：

两个正整数n,sumn,sum。

输出格式：

输出包括11行，为字典序最小的那个答案。

当无解的时候，请什么也不输出。（好奇葩啊）

输入输出样例

输入样例#1：复制

4 16

输出样例#1：复制

3 1 2 4

说明

对于40\%40%的数据，n≤7n≤7；

对于80\%80%的数据，n≤10n≤10；

对于100\%100%的数据，n≤12,sum≤12345n≤12,sum≤12345。

一开始用模拟法超时3个点

然后观察发现累合的乘数为杨辉三角用杨辉三角优化超时2个点

#include<bits/stdc++.h>
using namespace std;
//input
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i>=(b);i--)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m);
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define ll long long
#define inf 0x3f3f3f3f
#define REP(i,N)  for(int i=0;i<(N);i++)
#define CLR(A,v)  memset(A,v,sizeof A)
//
#define N 105
int a[16];
int yhsj[15][15];
int main()
{int n;int sum;RII(n,sum);rep(i,1,n)a[i]=i;yhsj[1][1]=1;rep(i,2,n)rep(j,1,i)yhsj[i][j]=yhsj[i-1][j-1]+yhsj[i-1][j];int b[16];do{int all=0;int ok=1;rep(i,1,n){all+=a[i]*yhsj[n][i];if(all>sum){ok=0;break;}}if(ok&&all==sum){rep(i,1,n){if(i!=1)printf(" ");printf("%d",a[i]);}break;}}while(next_permutation(a+1,a+1+n));return 0;
}

2 TLE

参考了大佬的做法

其实只要加一个关键剪枝即可

如果加到i处过不去了把i及其后面的数降序排列好下一个next就是累合杨辉三角的最小值了！！！！（因为杨辉三角中间大两边小）

#include<bits/stdc++.h>
using namespace std;
//input
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i>=(b);i--)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m);
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define ll long long
#define inf 0x3f3f3f3f
#define REP(i,N)  for(int i=0;i<(N);i++)
#define CLR(A,v)  memset(A,v,sizeof A)
//
#define N 105
int a[16];
int yhsj[15][15];
int main()
{int n;int sum;RII(n,sum);rep(i,1,n)a[i]=i;yhsj[1][1]=1;rep(i,2,n)rep(j,1,i)yhsj[i][j]=yhsj[i-1][j-1]+yhsj[i-1][j];do{int all=0;int ok=1;rep(i,1,n){all+=a[i]*yhsj[n][i];if(all>sum){ok=0;sort(a+i,a+1+n,greater<int>());   break;}}if(ok&&all==sum){rep(i,1,n){if(i!=1)printf(" ");printf("%d",a[i]);}break;}}while(next_permutation(a+1,a+1+n));return 0;
}

View Code

其实这题用dfs回溯法更见简单高效上面那个剪枝其实很难想到

不要过度依赖STL 有时候效率非常低下

#include<bits/stdc++.h>
using namespace std;
//input
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i>=(b);i--)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m);
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define ll long long
#define inf 0x3f3f3f3f
#define REP(i,N)  for(int i=0;i<(N);i++)
#define CLR(A,v)  memset(A,v,sizeof A)
//
#define N 105int a[16];
int yhsj[15][15];
int vis[16];
int sum,n;
int ok=0;
int ans[16];
void dfs(int now,int all)
{if(ok)return ;if(now==n+1&&all==sum){ok=1;rep(i,1,n){if(i!=1)printf(" ");printf("%d",ans[i]);}return ;}rep(i,1,n){if(vis[i])continue;if(all+i*yhsj[n][now]>sum)continue;vis[i]=1;ans[now]=i;dfs(now+1,all+i*yhsj[n][now]);vis[i]=0;}return ;
}
int main()
{RII(n,sum);rep(i,1,n)a[i]=i;yhsj[1][1]=1;rep(i,2,n)rep(j,1,i)yhsj[i][j]=yhsj[i-1][j-1]+yhsj[i-1][j];dfs(1,0);return 0;
}

View Code

转载于:https://www.cnblogs.com/bxd123/p/10664413.html