2022“杭电杯”中国大学生算法设计超级联赛（4） Magic

#include<bits/stdc++.h>

using namespace std;

#define X first
#define Y second
#define eps  1e-2
#define gcd __gcd
#define pb push_back
#define PI acos(-1.0)
#define lowbit(x) (x)&(-x)
#define bug printf("!!!!!\n");
#define mem(x,y) memset(x,y,sizeof(x))

typedef long long LL;
typedef long double LD;
typedef pair<int,int> pii;
typedef unsigned long long uLL;

#define int long long

const int maxn = 1e5+2;
const int INF  = 1<<30;
const int mod  = 1e9+7;

struct edge {
  int v, w;
};

vector<edge> e[maxn];
int dis[maxn], cnt[maxn], vis[maxn];
queue<int> q;

inline void addedge(int u, int v, int w) {  // 加边
  e[u].push_back({v,w});
}

int n,k,a[maxn];
bool spfa(int n, int s) {
  // memset(dis, 63, sizeof(dis));
  for(int i=0;i<=n;i++) dis[i]=1e9,vis[i]=cnt[i]=0;
  while(q.size()) q.pop();
  dis[s] = 0, vis[s] = 1;
  q.push(s);
  while (!q.empty()) {
    int u = q.front();
    // cout<<u<<endl;
    q.pop(), vis[u] = 0;
    for (auto ed : e[u]) {
      int v = ed.v, w = ed.w;
      if (dis[v] > dis[u] + w) {
        dis[v] = dis[u] + w;
        cnt[v] = cnt[u] + 1;  // 记录最短路经过的边数
        if (cnt[v] >= n){
          // cout<<cnt[v]<<" "<<"NO"<<" "<<v<<endl;
           return false;
        }
        // 在不经过负环的情况下，最短路至多经过 n - 1 条边
        // 因此如果经过了多于 n 条边，一定说明经过了负环
        if (!vis[v]) q.push(v), vis[v] = 1;
      }
    }
  }
  // cout<<"YES"<<endl;
  return true;
}

void solve(){
    scanf("%lld%lld",&n,&k);
    for(int i=0;i<=n+1;i++) e[i].clear();
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    for(int i=1;i<=n;i++){
      addedge(i,i-1,0);
      int L=max(0LL,i-k),R=min(n,i+k-1);
      addedge(R,L,-a[i]);
      // cout<<L<<" "<<R<<endl;
    }
    int q;
    scanf("%lld",&q);
    while(q--){
      int L,R,B;
      scanf("%lld%lld%lld",&L,&R,&B);
      addedge(L-1,R,B);
    }
    for(int i=0;i<=n;i++) addedge(n+1,i,0);
    if(spfa(n+2,n+1)==0){
      cout<<-1<<endl;return;
    }
    int res=1e9;
    for(int i=0;i<=n;i++) {
      // cout<<dis[i]<<" ";cout<<endl;
      res=min(res,dis[i]);
    }
    cout<<-res<<endl;
    return;
}

int32_t main()
{
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
//    ios::sync_with_stdio(false);
    int t = 1;
    scanf("%lld",&t);
    while(t--){
    //    printf("Case %d: ",cas++);
        solve();
    }
    return 0;
}