E-小雨的矩阵
题目描述
小雨有一个n×n 的矩阵,起点在(1,1),终点在(n,n),只能向下或向右走,且每次只能走 1 步。
矩阵上每个点都有一个点权a(i,j)。
求走到终点的路径有多少不同的点权和。
输入描述:
第一行,输入一个正整数 n 。
接下来 n+1 行,每行 n 个数,表示 a(i,j)
输出描述:
共一行,输出有多少不同的点权和。
示例1
输入
2
1 5
2 4
输出
2
说明
(1,1)→(2,1)→(2,2):和为7。
(1,1)→(1,2)→(2,2):和为10。
备注:
1 ≤ n ≤ 8, 0 ≤ ai,j ≤ 50
详解思路:
dfs模板题,dfs的第三个参数为点权和
#include<bits/stdc++.h>
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int maxn = 51;
int mp[maxn][maxn];
int vis[maxn][maxn];
set<int> s;
int dx[2] = {1,0};
int dy[2] = {0,1};
int temp;
int n;
void dfs(int x, int y, int sum) {
if (x == n && y == n) {
s.insert(sum);
return;
}
for (int i = 0; i < 2; i++) {
int tx = x+dx[i];
int ty = y+dy[i];
if ((tx>=1&&tx<=n) && (ty>=1&&ty<=n) && !vis[tx][ty]) {
vis[tx][ty] = 1;
dfs(tx, ty, sum+mp[tx][ty]);
vis[tx][ty] = 0;//回溯
}
}
}
int main() {
scanf("%d", &n);
memset(vis, 0, sizeof(vis));
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
scanf("%d", &mp[i][j]);
dfs(1,1,mp[1][1]);
cout << s.size();
}