Infinite Fraction Path（bfs剪枝）

2017ICPC沈阳

Problem

给出n个城市，编号为$[0,n-1]$，第i个城市有一条通向 $(i^2+1)\%n$ 的单向边。每个城市对应一个 0 ~ 9 的数字。求经过n个城市所组成的长为n的最大字典序路径。

Solution

为保证最终结果最大，应该尽量保证前面的数字尽量的大。所以，

1. 取出最大数字的下标，作为下一次的扩展边
2. 统计这些下标中指向的下一个数字中最大的
3. 开一个数组，统计每一位最大的数字

但仅有上面的这些是不够的。
假如，D里面每一个数字都是相同的，则最终的复杂度会上升到 $O(n^2)$。
为此，需要设置一个标记数组vis，对于扩展了相同长度的结点且下一步指向的边相同的则应该 “剪掉”,来优化时间。

Code

const int N = 2e5 + 5, M = 1e6 + 7;
int n;
string s;
int a[N], b[N], vis[N];
int maxn;
void bfs()
{
	queue<pii> q;
	for (int i = 0; i < n; i++)
		if (s[i] == maxn + '0')
			q.push({ i, 1 });
	b[0] = maxn;

	while (sz(q))
	{
		pii x = q.front(); q.pop();
		if (x.sd == n) continue;
		
		if (s[x.ft] == b[x.sd - 1] + '0')
		{
			if (vis[x.ft] == x.sd) continue;//剪枝
			vis[x.ft] = x.sd;

			if (s[a[x.ft]] - '0' >= b[x.sd])
				b[x.sd] = s[a[x.ft]] - '0',
				q.push({ a[x.ft], x.sd + 1 });
		}
	}
}

int main()
{
	IOS;
	int T, cas = 0; cin >> T;
	while(T--)
	{
		cout << "Case #" << ++cas << ": ";
		cin >> n >> s;
		for (int i = 0; i < n; i++)
			a[i] = ((ll)i * i + 1) % n;//记录指向的边
		
		//初始化
		maxn = 0;
		for (int i = 0; i < n; i++)
			b[i] = 0, vis[i] = 0;
		for (int i = 0; i < sz(s); i++)
			maxn = max(maxn, s[i] - '0');

		bfs();
		for (int i = 0; i < n; i++)
			cout << b[i];
		cout << "\n";
	}


	return 0;
}