查了查题解,是用的递推式。
我用了dfs。自己测试的时候很慢。但是竟然AC了。
#include <stdio.h>
int n[40];
long long count;
int k;
// 1 2 3 -- color
void dfs(int i)
{
if (i > k)
{
count++;
// test
/*for (int p = 1; p <= k; p++)
printf("%d ", n[p]);
printf("\n");*/
return;
}
if (1 == i || 2 == i)
{
for (int j = 1; j <= 3; j++)
{
n[i] = j;
dfs(i + 1);
}
}
else
{
for (int j = 1; j <= 3; j++)
{
if ((j == n[i - 1] || j == n[i - 2]) || (n[i - 1] == n[i - 2]))
{
n[i] = j;
dfs(i + 1);
}
}
}
}
int main()
{
int c;
scanf("%d", &c);
while (c--)
{
scanf("%d", &k);
count = 0;
dfs(1);
printf("%lld\n", count);
}
return 0;
}