CF576D Flights for Regular Customers 矩阵乘法 + Bitset优化

%%%cxhscst2's blog

Codeforces 576D Flights for Regular Customers（矩阵加速DP）

代码非常优美 + 简洁，学习到了

Code:

#include <bits/stdc++.h>
#define N 160
#define inf 0x3f3f3f3f
#define maxn 1000000
#define setIO(s) freopen(s".in","r",stdin)
using namespace std;
int n,m,ans=inf;
int dis[N][N], f[N][N];
int cnt;
bitset<N> can[N], now[N], tmp[N], base[N];
struct Node
{
	int x,y,d;
	void scan() { scanf("%d%d%d",&x,&y,&d); }
	friend bool operator < (const Node &a, const Node &b)
	{
		return a.d < b.d;
	}
}e[N];
void Mul(bitset<N>*a,bitset<N>*b)
{
	bitset<N>ret[N];
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j) if(a[i][j]) ret[i]|=b[j];
	for(int i=1;i<=n;++i) a[i]=ret[i];
}
void Pow(bitset<N>*a,int b)
{
	bitset<N>ret[N];
	for(int i=1;i<=n;++i) ret[i][i]=1;
	for(;b;b>>=1)
	{
		if(b&1) Mul(ret, a);
		Mul(a,a);
	}
	for(int i=1;i<=n;++i) a[i]=ret[i];
}
int main()
{
	// setIO("input");
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;++i) e[i].scan();
	sort(e+1,e+1+m);
    for(int i=1;i<=n;++i) for(int j=1;j<=n;++j) dis[i][j]=inf;
    for(int i=1;i<=n;++i) dis[i][i]=0;
    cnt=0,ans=inf;
    for(int i=1;i<=n;++i) can[i][i]=1;
    for(int i=1;i<=m;++i)
    {
    	int x=e[i].x,y=e[i].y,d=e[i].d;
    	for(int j=1;j<=n;++j) tmp[j]=base[j];
    	Pow(tmp, d-cnt);
        Mul(can,tmp);
        for(int j=1;j<=n;++j) for(int k=1;k<=n;++k) dis[j][k]=min(dis[j][k], dis[j][x]+1+dis[y][k]);
        for(int j=1;j<=n-1;++j) if(can[1][j]) ans=min(ans, d + dis[j][n]);
        cnt=d;
        base[x][y]=1;
    }
    if(ans<0x3f3f3f3f) printf("%d\n",ans);
    else printf("Impossible\n");
	return 0;
}