11.动态规划入门
A.一维递推
1.错排问题
错排问题+递推问题
某君写了n封信,有n个信封,如果所有的信都装错了信封,那么会有多少种情况?
#include<iostream>
using namespace std;
const int N=1e3+9;
long long f[N];
int main(){
int n;
cin>>n;
//方法一:数组存储
f[1]=0,f[2]=1;
for(int i=3;i<=n;i++){
f[i]=(f[i-1]+f[i-2])*(i-1);
}
cout<<f[n]<<endl;
//方法二:空间优化
long long a=0,b=1,c=1;
for(int i=3;i<=n;i++){
c=(a+b)*(i-1);
a=b;
b=c;
}
if(n!=1){a
cout<<c<<endl;
}else{
cou<<0<<endl;
}
}
2.斐波那契数列
/*斐波那契数列*/
#include<iostream>
using namespace std;
const int N=1e3;
long long f[N];
int main(){
int n;
cin>>n;
//方法一:用数组存储中间变量,占用空间稍大
f[0]=f[1]=1;
for(int i=2;i<=n;i++){
f[i]=f[i-1]+f[i-2];
}
cout<<f[n]<<endl;
//方法二:持续更新,空间小,无法存储中间变量
long long a=1,b=1,c=1;
for(int i=2;i<=n;i++){
c=a+b;//i
a=b;//i-2
b=c;//i-1
}
cout<<c<<endl;
}
B.二维递推
1.杨辉三角
#include<iostream>
using namespace std;
int n;
int a[10][10]= {0};
int main() {
cin>>n;
a[0][0]=1;
for(int i=1; i<n; i++) {
a[i][0]=1;
for(int j=1; j<n; j++) {
a[i][j]=a[i-1][j-1]+a[i-1][j];
}
}
for(int i=0; i<n; i++) {
for(int j=0; j<n; j++) {
cout<<a[i][j]<<" ";
}
cout<<endl;
}
int m;
cin>>m;
cout<<a[n-1][m-1]<<endl;
return 0;
}
/*
5
1 0 0 0 0
1 1 0 0 0
1 2 1 0 0
1 3 3 1 0
1 4 6 4 1
3
6
*/
2.过河卒
f[0][0]=1;
for(int i=0; i<=n; i++) {
for(int j=0; j<=m; j++) {
if(i!=0) {
f[i][j]=f[i][j]+f[i-1][j];
}
if(j!=0) {
f[i][j]=f[i][j]+f[i][j-1];
}
}
}
//f[n][m]即为从(0,0)到(n,m)的路径数目
int cx,cy;
int x[8]={1,1,2,2,-1,-1,-2,-2};//横向位移
int y[8]={2,-2,1,-1,2,-2,1,-1};//纵向位移
int d[30][30];//用来记录是否是马控制点
for(int i=0;i<30;i++){//将 d 数组转化为 0
for(int j=0;j<30;j++){
d[i][j]=0;
}
}
d[cx][xy]=1;//用 1 来表示是马的控制点
for(int i=0;i<8;i++){
int tx=cx+x[i];//计算马控制点坐标
int ty=cy+y[i];
if(tx>=0 && tx<=n && ty>=0 && ty<=n){
d[tx][ty]=1;//记录为马控制点
}
}
#include<iostream>
using namespace std;
int dir[8][2]= {{1,2},{1,-2},{2,1},{2,-1},{-1,2},{-1,-2},{-2,1},{-2,-1}};
bool d[30][30];
long long dp[30][30];
int main() {
int n,m,cx,cy;// n,m棋盘大小 cx,xy马的位置
cin>>n>>m>>cx>>cy;
d[cx][cy]=true;
for(int i=0; i<8; i++) {
int tx=cx+dir[i][0];
int ty=cy+dir[i][1];
if(tx>=0 && tx<=n && ty>=0 && ty<=n) {
d[tx][ty]=true;//记录为马控制点
}
}
dp[0][0]=1;//dp[n][m]即为从(0,0)到(n,m)的路径数目
for(int i=0; i<=n; i++) {
for(int j=0; j<=m; j++) {
if(d[i][j]==false) {
if(i) {
dp[i][j]+=dp[i-1][j];
}
if(j) {
dp[i][j]+=dp[i][j-1];
}
}
}
}
cout<<dp[n][m]<<endl;
return 0;
}
/*
样例输入:
5 5 2 4
样例输出:
14
*/
C.动态规划入门/DP
1.最短路径
/*输入一张地图,求从左上角到右下角的最短路径长,只能往右/下方向移动*/
#include<iostream>
#include<algorithm>
using namespace std;
int a[110][110];
int dp[110][110];
int main(){
int n;
cin>>n;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
cin>>a[i][j];
}
}
dp[1][1]=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
if(i==1 && j==1){
continue;
}else if(i==1){
dp[i][j]=dp[i][j-1]+a[i][j];
}else if(j==1){
dp[i][j]=dp[i-1][j]+a[i][j];
}else{
dp[i][j]=min(dp[i-1][j],dp[i][j-1])+a[i][j];
}
}
}
cout<<dp[n][n]<<endl;
return 0;
}
/*
样例输入:
3
0 3 4
6 2 5
5 4 3
样例输出:
12
*/
2.捡水果
#include<iostream>
#include<algorithm>
using namespace std;
int n;
int a[105][105]={0};
int dp[105][105]={0};
int main() {
cin>>n;
for(int i=1; i<=n; i++) {
for(int j=1; j<=i; j++) {
cin>>a[i][j];
}
}
int ans=0;
for(int i=1; i<=n; i++) {
for(int j=1; j<=i; j++) {
dp[i][j]=a[i][j]+max(dp[i-1][j],dp[i-1][j-1]);
if(i==n){
ans=max(dp[i][j],ans);
}
}
}
cout<<ans<<endl;
cout<<"--输出dp数组--"<<endl;
for(int i=1; i<=n; i++) {
for(int j=1; j<=i; j++) {
cout<<dp[i][j]<<" ";
}
cout<<endl;
}
return 0;
}
/*
样例输入:
4
3
1 2
6 2 3
3 5 4 1
样例输出:
15
--输出dp数组--
3
4 5
10 7 8
13 15 12 9
*/
3.蒜头君回家/三维
/*蒜头君回家 < 进阶--三维坐标版 > */
/*假设蒜头君存在于三维空间中,他想要从 (0,0,0)回到家(n,n,n),她每走一步
都会消耗当期坐标 a[i][j][k] 的体力,问他至少消耗多少体力才能回家*/
#include<iostream>
#include<algorithm>
using namespace std;
// 状态方程:
// f[i][j][k] = f[i][j][k] + min ( f[i-1][j][k] + f[i][j-1][k] + f[i][j][k-1] )
int f[105][105][105];
int inf=10000;
int main(){
int x,y,z;
cin>>x>>y>>z;
for(int i=0;i<=x;i++){
for(int j=0;j<=y;j++){
for(int k=0;k<=z;k++){
cin>>f[i][j][k];
}
}
}
for(int i=0;i<=x;i++){
for(int j=0;j<=y;j++){
for(int k=0;k<=z;k++){
int mi=inf;
if(i!=0){
mi=min(mi,f[i-1][j][k]);
}
if(j!=0){
mi=min(mi,f[i][j-1][k]);
}
if(k!=0){
mi=min(mi,f[i][j][k-1]);
}
if(mi!=inf){
f[i][j][k]+=mi;
}
}
}
}
cout<<f[x][y][z]<<endl;
return 0;
}
/*
样例输入:
2 2 2 //每一维坐标的宽(实际为3)
1 2 3 // 一维坐标
4 5 6
7 8 9
1 2 3 // 二维
4 5 6
7 8 9
1 2 3 // 三维
4 5 6
7 8 9
样例输出:
23
*/
4.爬楼梯
#include<iostream>
using namespace std;
int n;
int mod=1e5+7;
int dp[1005];
int main(){
cin>>n;
dp[1]=1,dp[2]=2;
for(int i=3;i<=n;i++){
dp[i]=(dp[i-1]+dp[i-2])%mod;
}
cout<<dp[n]<<endl;
return 0;
}
#include<iostream>
using namespace std;
const int mod=1e5+7;
int dp[1005];
int main(){
int n;
cin>>n;
dp[0]=1;
for(int i=1;i<=n;i++){
dp[i]=0;
for(int j=i-1;j>=0;j=j-2){
dp[i]=dp[i]+dp[j];
dp[i]=dp[i]%mod;
}
}
cout<<dp[n]<<endl;
return 0;
}
/*
输入:4
输出:3
*/
5.传球游戏
/*传球游戏:共有n个人围成一圈,从A开始传球,传递m次,
最后回到A的传球方法有几种(接球顺序不同视为不同的方法)*/
#include<iostream>
#include<cstring>
using namespace std;
int f[35][35];
int main(){
int n,m;
cin>>n>>m;
memset(f,0,sizeof(f));
f[0][1]=1; // f【传了多少次】【现在在谁手上】= 的方法数
for(int i=1;i<=m;i++){
for(int j=1;j<=n;j++){
if(j==1){
f[i][j]=f[i-1][2]+f[i-1][n];
}else if(j==n){
f[i][j]=f[i-1][1]+f[i-1][n-1];
}else{
f[i][j]=f[i-1][j-1]+f[i-1][j+1];
}
}
}
cout<<f[m][1]<<endl;
cout<<endl;
}
/*
cin:
3 3
cout:
2
*/
6.一维消消乐
一维消消乐是一款非堂筒单的游戏。有几颗一排,每一颗珠子有一个价值(可能是负的)。
游戏规则是这样,你可以选择如若干对相邻的珠子,让他们同时消去,每一对珠子的消失,都会使得总分数加上两珠子相乘的分数。注意到每个珠子只能消一次,而且珠子消去以后,还会占位。
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=10005;
int dp[maxn][2];//1代表和前面的组合,0代表没有和前面的组合
int w[maxn];
int main(){
int n;
cin>>n;
for(int i=1;i<=n;i++){
cin>>w[i];
}
dp[1][0]=0;
for(int i=2;i<=n;i++){
dp[i][0]=max(dp[i-1][0],dp[i-1][1]);
dp[i][1]=dp[i-1][0]+w[i]*w[i-1];
}
cout<<max(dp[n][0],dp[n][1])<<endl;
return 0;
}
/*
cin:
8
-9 -5 -4 -2 4 -5 -4 2
cout:
73
*/
7.数组分组
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=1010;
const int mod=1000;
int a[maxn];
int dp[maxn];
int pre[maxn][maxn];
int main(){
int n;
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i];
}
for(int i=1;i<=n;i++){
pre[i][i]=a[i];
for(int j=i+1;j<=n;j++){
pre[i][j]=pre[i][j-1]*a[j]%mod;
}
}
dp[0]=0;
dp[1]=a[1];
for(int i=2;i<=n;i++){
for(int j=0;j<i;j++){
dp[i]=max(dp[i],dp[j]+pre[j+1][i]);
}
}
cout<<dp[n]<<endl;
return 0;
}
/*
cin:
7
52 26 1 36 72 48 43
cout:
1596
*/
8.墙壁涂色
#include<iostream>
using namespace std;
int n;
int a[100];
int main() {
cin>>n;
a[1]=3;//一块墙
a[2]=6;
a[3]=6;
for(int i=4; i<=n; i++) {
a[i]=a[i-1]+a[i-2]+a[i-2];
//第1面墙与第i-1面墙颜色不同,第i面墙有a[i-2]种
//第1面墙与第i-1面墙颜色相同,第i面墙有a[i-2]种
}
cout<<a[n]<<endl;
return 0;
}
/*
cin:
4
cout:
18
*/
9.过河
在一个夜黑风高的晚上,有n(n<=50)个小朋友在桥的一边,由于桥很窄,每次只允许不大于2人通过。他们只有一个手电筒,所以每次过桥的两人需要把手电筒再带回来。i 号小朋友过桥的时间为T[i],两人过桥的总时间为二者中时间长者。问所有小朋友过桥的总时间最短是多少?解答思路
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n;
int a[1005];
int f[1005];//多少人过去用时,且手电筒在对面
int main(){
cin>>n;
for(int i=0;i<n;i++){
cin>>a[i];
}
sort(a,a+n);//升序
f[0]=a[0];
f[1]=a[1];
for(int i=2;i<n;i++){
f[i]=min(f[i-1]+a[0]+a[i],f[i-2]+a[0]+2*a[1]+a[i]);
}
cout<<f[n-1]<<endl;
return 0;
}
/*
cin:
4
1 5 6 7
cout:
20
*/