一、
话说大诗人李白,一生好饮。幸好他从不开车。
一天,他提着酒壶,从家里出来,酒壶中有酒 2 斗。他边走边唱:
无事街上走,提壶去打酒。
逢店加一倍,遇花喝一斗。
这一路上,他一共遇到店 N 次,遇到花 M 次。已知最后一次遇到的是花, 他正好把酒喝光了。
请你计算李白这一路遇到店和花的顺序,有多少种不同的可能?
注意:壶里没酒 ( 0 斗) 时遇店是合法的,加倍后还是没酒;但是没酒时遇花是不合法的。
输入格式
第一行包含两个整数 N 和 M.
输出格式
输出一个整数表示答案。由于答案可能很大,输出模 1000000007 的结果。
样例输入
5 10
样例输出
14
方法一
剪枝(去除不符合的情况)
dp dfs(深度优先搜索算法)
回溯算法
#include<stdio.h>
#include<string.h>
#define mod 1000000007
int N, M, flag = 0, k = 2, i = 1, arr[105];//M是总花束N是总店数k是酒的余量 arr为0遇到花 arr为1遇到店
void dfs(int x, int y)
{
if (x>N || y>M) //如果超出去就剪枝
{
return;
}
if (x<N&&y == M) //花遇见完了,但是还有店没有遇见完全就剪枝
{
return;
}
if (x == N&&M - y != k) //店遇见完了,但是后面剩的花不足以将酒喝完就剪枝
{
return;
}
if (k == -1 && arr[i - 1] == 0) //判断没有酒了但是还是遇见花,这是不合理的
{
return;
}
if (x == N&&y == M&&k == 0 && arr[i - 1] != 1) //这里是判断最后一次不能遇到店,只能遇到花//
{
flag = (flag + 1) % mod;
return;
}
arr[i] = 0; //遇到花了
k--;
i++;
dfs(x, y + 1);
i--;
k++;
arr[i] = 1; //遇到店了
k *= 2;
i++;
dfs(x + 1, y);
i--;
k /= 2;
}
int main()
{
while (1)
{
memset(arr, 0, sizeof(arr));
scanf("%d%d", &N, &M);
dfs(0, 0);
printf("%d", flag%mod);
}
return 0;
}
但是会超时(因为这个题暴搜的时间复杂度是o(2n))
方法二 三维数组动态规划
我们可以讲这个题看成是一个二维的矩阵,来走方格;
用题目中给的例子,5个店,10个花
向下走我们此时手里面的酒就会翻倍,向右走我们手中的酒就会减一
所以刚开始我们手中的酒就是2然后我们必须保证走到最后一格格子剩下的酒必须是0,而且我们最后一步必须遇到的是花,所以如果只是开二维的话会存在后效性,我们必须开到三维才可以,所以第三维我们记录现在所含有的酒就好了
@:从8那个点开始后面的就不需要算了,因为假如你再遇到店,那么你此时手里面的酒就是16个,你总共的花才10个所以就不可能喝完了,所以我们就直接不用算下面的了
.......:表示不用算了
但是这样算有个不能保证最后一步必须是遇到花了,所以我们从终点往起点走就好了.
所以我们可以写出递推式:
一遇见花了:dp[i][j-1][k-1]+=dp[i][j][k];
二遇见店了:dp[i-1][j][k<<1]+=dp[i][j][k];
#include<stdio.h>
#include<string.h>
#define mod 1000000007
#define max 105
int dp[max][max][max];
int main()
{
int N, M, flag = 0, i, j, k;
scanf("%d%d", &N, &M);
memset(dp, 0, sizeof(dp));
dp[N][M][2] = 1; //初始化刚开始手里面是有2斗酒 店是N花是M
for (i = N; i >= 0; i--)
{
for (j = M; j >= 0; j--)
{
for (k = M; k >= 0; k--)
{
if (2 * k <= M&&i >= 1)
{
dp[i - 1][j][k << 1] = (dp[i][j][k] + dp[i - 1][j][k << 1]) % mod;
}
if (j >= 1 && k >= 1)
{
dp[i][j - 1][k - 1] = (dp[i][j - 1][k - 1] + dp[i][j][k]) % mod;
}
}
}
}
printf("%d", dp[0][1][1] % mod);
return 0;
}
二、
题目描述
小蓝有一个长度为 n 的数组 A = (a1, a2, · · · , an),数组的子数组被定义为从原数组中选出连续的一个或多个元素组成的数组。数组的最大公约数指的是数组中所有元素的最大公约数。如果最多更改数组中的一个元素之后,数组的最大公约数为 g,那么称 g 为这个数组的近似 GCD。一个数组的近似 GCD 可能有多种取值。
具体的,判断 g 是否为一个子数组的近似 GCD 如下:
1. 如果这个子数组的最大公约数就是 g,那么说明 g 是其近似 GCD。
2. 在修改这个子数组中的一个元素之后(可以改成想要的任何值),子数组的最大公约数为 g,那么说明 g 是这个子数组的近似 GCD。
小蓝想知道,数组 A 有多少个长度大于等于 2 的子数组满足近似 GCD 的值为 g。
输入格式
输入的第一行包含两个整数 n, g,用一个空格分隔,分别表示数组 A 的长度和 g 的值。
第二行包含 n 个整数 a1, a2, · · · , an,相邻两个整数之间用一个空格分隔。
输出格式
输出一行包含一个整数表示数组 A 有多少个长度大于等于 2 的子数组的近似 GCD 的值为 g 。
样例输入
5 3 1 3 6 4 10
样例输出
5
提示
满足条件的子数组有 5 个:
[1, 3]:将 1 修改为 3 后,这个子数组的最大公约数为 3 ,满足条件。
[1, 3, 6]:将 1 修改为 3 后,这个子数组的最大公约数为 3 ,满足条件。
[3, 6]:这个子数组的最大公约数就是 3 ,满足条件。
[3, 6, 4]:将 4 修改为 3 后,这个子数组的最大公约数为 3 ,满足条件。
[6, 4]:将 4 修改为 3 后,这个子数组的最大公约数为 3,满足条件。
对于 20% 的评测用例,2 ≤ n ≤ 102 ;
对于 40% 的评测用例,2 ≤ n ≤ 103;
对于所有评测用例,2 ≤ n ≤ 105 , 1 ≤ g, ai ≤ 109。
1.(时间超限)
#include<stdio.h>
#include<string.h>
int n = 0, g = 0;
int nu = 0;
int num[10000000];
int main()
{
while (1)
{
scanf("%d %d", &n, &g);
memset(num, 0, sizeof(num));
for (int i = 0; i < n; i++)
{
int t;
scanf("%d", &t);
if ((t%g) == 0)
{
num[i] = 1;
}
}
for (int i = 0; i < n; i++)
{
printf("%d\n", num[i]);
}
int number = 0;
for (int i = 0; i < n; i++)
{
nu = 0;
if (num[i] == 1)
{
for (int j = i+1; j < n; j++)
{
if (num[j] == 0)
{
nu = nu + 1;
//printf("%d\n", number);
}
if (nu <= 1)
{
number++;
//printf("%d2", nu);
//printf("%d\n", number);
}
}
}
else if (num[i] == 0)
{
for (int j = i+1; j < n; j++)
{
if (num[j] == 0)
{
nu = 1;
//printf("%d\n", number);
}
if (nu ==0)
{
number++;
//printf("%d", nu);
/*printf("%d\n", number);*/
}
}
}
//printf("\n");
}
printf("%d\n", number);
}
return 0;
}
2.满分题解
#include <iostream>
#include <cstring>
const int N = 1e5 + 10;
int n, g;
bool st[N];
int main(int argc, char** argv) {
scanf("%d%d", &n, &g);
memset(st, 0, sizeof st);
// 预处理是否是 g 的合数
for(int i = 1; i <= n; ++ i) {
int t;
scanf("%d", &t);
if(t % g == 0) st[i] = true;
}
int len_g = 0, len_ung = 0;
// len_g 表示连续是g的合数的长度,len_ung 含有一个非g合数的子数组的长度
long long res = 0;
// 依次枚举以 ai 为结尾的答案的数目
// 并维护 需要的数组长度
for(int i = 1; i <= n; ++ i) {
if(st[i]) {
res += len_ung;
len_g ++;
len_ung ++;
}
else {
res += len_g;
len_ung = len_g + 1;
len_g = 0;
}
}
printf("%lld\n", res);
return 0;
}