2021百度之星初赛一 1003.鸽子(小dp)

Posted 2021-08-31 issue是fw

LINK

考虑定义$f[i][j]$表示考虑$[1,i]$的指令,现在坏电脑在$j$位置需要忽略的最少操作

如果忽略这条指令显然有$f[i][j]=f[i-1][j]+1$

如果不忽略这条指令

考虑$j==l_i$，有$f[i][j]=f[i-1][l_i]$

考虑$j==r_i$,有$f[i][j]=f[i-1][r_i]$

否则,$f[i][j]=f[i-1][j]$

也就是每次转移,对于$j!=l_i\&\&j!=r_i$的状态直接保留就好了,对于其他两个状态额外转移即可

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+10;
const int inf = 1e9;
int l[maxn],r[maxn],f[maxn],n,m,k;
int main()
{
	int t; cin >> t;
	while( t-- )
	{
		cin >> n >> m >> k;
		for(int i=1;i<=n;i++)	f[i] = inf;
		for(int i=1;i<=m;i++)	scanf("%d%d",&l[i],&r[i] );
		f[k] = 0;
		for(int i=1;i<=n;i++)
		{
			int L = min( f[r[i]]+1, f[l[i]] ), R = min( f[l[i]]+1, f[r[i]] );
			f[l[i]] = R; f[r[i]] = L;
		}
		for(int i=1;i<=n;i++)
		{
			if( f[i]==inf )	f[i] = -1;
			printf("%d%c",f[i],i==n?'\\n':' ');
		}
	}
}