SMU Summer 2024 Team Round 1

Problem - L - Codeforces

题意：

        n个节点m条边的无向图中，有k个节点是大门，这些大门都有特定的开放时间，一个点到大门的花费为点的权值加上路径上的边权之和，求出1到T每一时刻所有节点到大门的最小花费之和。若某一时刻没有大门打开则输出-1

        简化为给定n个点m边的无向图，每个点和边都有权值，每一时刻都可能存在几个特殊点，求出每一时刻所有的点到其最小花费特殊点的总花费之和（其自身的花费为路径边权与自身权值之和）。

分析：

        n和m和T都是1e5，k是15。我们可以观察到大门的数量是非常少的，那么可以考虑从大门入手。仔细思考之后我们会发现，若某些时刻打开的大门类型是一样的，那么他们的花费其实就是一样的，并且这个大门的种类数最多不超过30，单独的算出来每一种类型需要多少花费的时间复杂度其实是O（n）的。那么我们怎么找出每种类型呢？我们遍历T和k，如果当前大门打开的时间在当前Ti内，那么我们就用一个vector<int>num存起来，把所有大门都遍历完之后，再把当前num存入map，这样map就会自动帮我们去重，并且还可以用键值对来记录每个类型的值。然后我们只需要遍历map和n，找出当前节点到当前打开的大门中最小的花费，那么这种类型的最小花费就算出来了，将map中的值修改。要找出当前节点到当前打开的大门中的花费，我们不需要每次都计算，我们可以用dijkstra提前预处理出每个大门到所有节点的最小花费，时间复杂度O（kmlogm）。这样我们就处理出了所有类型的最小花费。最后每一时刻，我们之前已经用num数组处理出了每一时刻的类型，并且用map存好了其花费，直接输出就好，但如果某一时刻num数组为空，说明没有大门打开，输出-1.

代码：

#include<bits/stdc++.h>
#define endl '\n'
#define mk make_pair
#define int long long
using namespace std;
typedef pair<int, int> PII;
const int N = 1e5+10;
const int mod = 998244353;
int n,m,k,l,T,u,v,w,cnt;
int a[N];
int h[N];
int jl[17][N]={LLONG_MAX};
struct node {
    int id,l,r;
}K[17];

struct edge{
    int to,next,w;
}bi[2*N];

void addedge(int u,int v,int w){
    ++cnt;
    bi[cnt].to=v;
    bi[cnt].w=w;
    bi[cnt].next=h[u];
    h[u]=cnt;
}

void dijkstra(int x){
    jl[x][K[x].id]=0;
    priority_queue<PII,vector<PII>,greater<PII> >q;
    q.push({0,K[x].id});
    bool vis[N];
    memset(vis,0,sizeof(vis));
    while(q.size()){
        auto u=q.top();q.pop();
        if(vis[u.second]) continue;
        vis[u.second]=1;
        //cout<<h[u.second]<<endl;
        for(int i=h[u.second];i!=-1;i=bi[i].next){
            if(jl[x][u.second]+bi[i].w<jl[x][bi[i].to]){
                jl[x][bi[i].to]=jl[x][u.second]+bi[i].w;
                q.push({jl[x][bi[i].to],bi[i].to});
            }
        }
    }
}

void sovle(){
    cin>>n>>m>>k>>T;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=k;i++) cin>>K[i].id>>K[i].l>>K[i].r;
    memset(h,-1,sizeof(h));
    for(int i=1;i<=k;i++){
        for(int j=1;j<=n;j++){
            jl[i][j]=LLONG_MAX;
        }
    }
    for(int i=1;i<=m;i++){
        cin>>u>>v>>w;
        addedge(u,v,w);
        addedge(v,u,w);
    }
    for(int i=1;i<=k;i++){
        dijkstra(i);
    }
    vector<int>num[N];
    map<vector<int>,int>v;
    for(int i=1;i<=T;i++){
        for(int j=1;j<=k;j++){
            if(K[j].l<=i&&K[j].r>=i){
                num[i].push_back(j);
            }
        }
        v.insert({num[i],0});
    }
    for(auto &ed:v){
        for(int i=1;i<=n;i++){
            int min1=LLONG_MAX;
            for(auto qd:ed.first){
                min1=min(min1,jl[qd][i]*a[i]);
            }
            ed.second+=min1;
        }
    }
    for(int i=1;i<=T;i++){
        if(!num[i].size()) cout<<-1<<endl;
        else cout<<v[num[i]]<<endl;
    }
}

signed main()
{	
    ios::sync_with_stdio(false), cin.tie(0),cout.tie(0); 
    int t = 1;
    //cin>>t;
    while (t--){
        sovle();
    }

    return 0;
}