假设一段时间内没有钟死亡,那么这段时间内每个钟的财产收益都不变,死亡时间也是确定的。
考虑一个钟死亡以后只会对其左右两边的钟的收益,以及死亡时间有影响,所以可以用一个set维护每个钟的死亡时间。每次取出时间最小的那个元素,然后更新它的左右两边。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<set>
#define ll long long
using namespace std;
const int maxn=1e6+10;
ll fight[110][110],l[maxn],r[maxn],w[maxn],t[maxn],cost[maxn],now;
int n,c,col[maxn];
struct clk
{
int x;
ll y;
};
std::set<clk>s;
std::set<clk>::iterator it;
operator <(clk a,clk b)
{
if(a.y==b.y) return a.x<b.x;
return a.y<b.y;
}
void cal(int x)
{
cost[x]=-fight[col[x]][col[l[x]]]-fight[col[x]][col[r[x]]];
}
ll gettime(int x)
{
if(w[x]-now*cost[x]<=0) return now;
if(cost[x]<=0) return 1e18;
return ((w[x]-1)/cost[x]+1);
}
void update(int x)
{
if(x<=0||x>n) return ;
s.erase((clk){x,gettime(x)});
w[x]-=cost[x]*now;
cal(x);
w[x]+=cost[x]*now;
s.insert((clk){x,gettime(x)});
}
int main()
{
cin>>c>>n;
for(int i=1;i<=c;i++)
for(int j=1;j<=c;j++)
scanf("%lld",&fight[i][j]);
for(int i=1;i<=n;i++)
scanf("%d",&col[i]);
for(int i=1;i<=n;i++)
{
l[i]=i-1;
r[i]=i+1;
w[i]=1;
t[i]=0;
cal(i);
s.insert((clk){i,gettime(i)});
}
now=0;
for(;(it=s.begin())->y<1e18;)
{
int i=it->x;
now=it->y;
s.erase(it);
l[r[i]]=l[i];
r[l[i]]=r[i];
update(l[i]);
update(r[i]);
}
printf("%d\n",col[s.begin()->x]);
return 0;
}
/*3 6
0 -1 1
1 0 -1
-1 1 0
1 2 3 1 2 3*/