EOJ 3297 铺瓷砖（DFS）

题目

http://acm.ecnu.edu.cn/problem/3297/

题意：

有一长度为 N(N≤30) 的地板，给定三种不同瓷砖：一种长度为1，一种长度为2，另一种长度为3，数目不限。要求将地板铺满，且任意两个相邻的瓷砖长度均不等。

求一共有多少种铺法？在所有的铺设方法中，一共用了长度为1的瓷砖多少块？

解题思路

在DFS的过程中，用参数n表示剩余地板长度，pre表示前一块瓷砖的长度，usedOne表示走到当前所使用的长度为1的瓷砖块数。

遍历三种长度的瓷砖，如果满足该瓷砖与前一块的长度不同且还铺得下，则铺下这块瓷砖，再者，如果铺下的这块瓷砖长度为1，则usedOne加1后再继续DFS。

AC代码

#include <bits/stdc++.h>
using namespace std;

int total_cnt = 0, one_cnt = 0;

//n表示剩余长度，pre表示前一个瓷砖长度，usedOne表示目前为止用了长度为1的瓷砖的个数
void dfs(int n, int pre, int usedOne)
{
if(n == 0) //铺完
{
total_cnt++;
one_cnt += usedOne;
return;
}
for (int i = 1; i <= 3; ++i)
{
if(i != pre && n >= i) //与前一个瓷砖不同且铺得下
{
if(i == 1) //使用长度为1的瓷砖
dfs(n-i, i, usedOne+1);
else //用长度为2或3的瓷砖
dfs(n-i, i, usedOne);
}
}
}

int main()
{
int T, n;
cin >> T;
while(T--)
{
cin >> n;
total_cnt = one_cnt = 0;
dfs(n, 0, 0);
cout << total_cnt << endl << one_cnt << endl;
}
return 0;
}