acwing第二场周赛(堆优化版Dj+最短路径树

给定一个整数 n，请你求出三元一次方程 3x+5y+7z=n 的一组非负整数解。

要求：

x≥0,y≥0,z≥0
如果解不唯一，则输出 x,y,z 字典序最小的解。
输入格式
第一行包含一个整数 T，表示共有 T 组测试数据。

每组数据占一行，包含一个整数 n。

输出格式
每组数据输出一行结果，如果无解则输出 −1，否则输出 x,y,z，整数之间单个空格隔开。

数据范围
对于前三个测试点，1≤n≤100。
对于全部测试点，1≤T≤1000，1≤n≤1000。
我写的三重循环代码：

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int x,y,z,t,n;
int main(){
    cin>>t;
    while(t--){
        
        cin>>n;
        for(int x=0;3*x<=n;x++){
            for(int y=0;3*x+5*y<=n;y++){
                for(int z=0;3*x+5*y+7*z<=n;z++){
                    if(3*x+5*y+7*z==n){cout<<x<<" "<<y<<" "<<z<<endl;goto in;}
                }
            }
        }
        cout<<"-1"<<endl;
        in:{}
    }
}

---

接下来是用两重循环替代掉三重循环的优化版代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int T,n;
void work(int x){
    for(int x=0;3*x<=n;x++){
        for(int y=0;3*x+5*y<=n;y++){
            int z=n-3*x-5*y;
            if(z%7==0){
                cout<<x<<" "<<y<<" "<<z/7<<endl;
                return;
            }
        }
    }
    puts("-1");
}
int main(){
    cin>>T;
    while(T--){
        cin>>n;
        work(n);
    }
}

更加简洁，更加简单明了，还是得多学啊胡耀文

---

给定一个由 n 个点和 m 条边组成的无向连通加权图。

设点 1 到点 i 的最短路径长度为 di。

现在，你需要删掉图中的一些边，使得图中最多保留 k 条边。

如果在删边操作全部完成后，点 1 到点 i 的最短路径长度仍为 di，则称点 i 是一个优秀点。

你的目标是通过合理进行删边操作，使得优秀点的数量尽可能大。

输入格式
第一行包含三个整数 n,m,k。

接下来 m 行，每行包含三个整数 x,y,w，表示点 x 和点 y 之间存在一条长度为 w 的边。

保证给定无向连通图无重边和自环。

输出格式
第一行包含一个整数 e，表示保留的边的数量 (0≤e≤k)。

第二行包含 e 个不同的 1∼m 之间的整数，表示所保留的边的编号。

按输入顺序，所有边的编号从 1 到 m。

你提供的方案，应使得优秀点的数量尽可能大。

如果答案不唯一，则输出任意满足条件的合理方案均可。

数据范围
对于前五个测试点，2≤n≤15,1≤m≤15。
对于全部测试点，2≤n≤105,1≤m≤105,n−1≤m,0≤k≤m,1≤x,y≤n,x≠y,1≤w≤109。

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#define int long long
using namespace std;
typedef pair<  pair<long long,int> , int> PII;
const int N = 200001;
int h[N],dist[N],w[N],e[N],idx,ne[N],st[N],n,m,k,a,b,c;
int res[N],cnt;
void add(int a,int b,int c){
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

void dj(){
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    priority_queue<PII,vector<PII>,greater<PII>>heap;
    heap.push({{0,1},-1});
    while(heap.size()){
        PII t=heap.top();
        heap.pop();
        int edge=t.second;//要存储的边
        int id=t.first.second;//编号
        if(st[id])continue;
        st[id]=1;
        res[cnt++]=edge;//记录答案
        for(int i=h[id];~i;i=ne[i]){
            int j=e[i];
            if(dist[j]>dist[id]+w[i]){
                dist[j]=dist[id]+w[i];
                heap.push(  {   {dist[j],j} , i});
                
            }
        }
    }
}

signed main(){
    cin>>n>>m>>k;
    memset(h, -1, sizeof h);
    while(m--){
        cin>>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
    }
    dj();
    int ans=min(cnt-1,k);
    cout<<ans<<endl;
    for(int i=1;i<=ans;i++){
        cout<<res[i]/2+1<<" ";
        //因为我们add（边）的时候是id=1的边idx记录了两次，所以1->1,2
        //2->3,4   3->5,6   知道m->2*m-1,2*m
        //所以反映射公式就是x=(y+1)/2《=》等价于x=y/2+1
    }
    
}

---

#include<bits/stdc++.h>
using namespace std;
const int N = 200010;
typedef long long LL;
typedef pair<LL,int>PII;
LL dist[N];
int h[N],e[N],ne[N],w[N],id[N],n,m,k,a,b,c,d,idx;
bool st[N];
vector<int>ans;

priority_queue<PII,vector<PII>,greater<PII>>heap;
void add(int a,int b,int c,int d){
    e[idx]=b,w[idx]=c,id[idx]=d ,ne[idx]=h[a],h[a]=idx++;
}
void dj(){
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    heap.push({0,1});
    while(heap.size()){
    auto t=heap.top();
    heap.pop();
    
    int ver=t.second;
    if(st[ver])continue;
    st[ver]=1;
    for(int i=h[ver];~i;i=ne[i]){
        int j=e[i];
        if(dist[j]>dist[ver]+w[i]){
            dist[j]=dist[ver]+w[i];
            heap.push({dist[j],j});
        }
    }
    
}
}
void dfs(int u)
{
    st[u] = true;
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (!st[j] && dist[j] == dist[u] + w[i])
        {
            if (ans.size() < k) ans.push_back(id[i]);
            dfs(j);
        }
    }
}

int main(){
    cin>>n>>m>>k;
    memset(h, -1, sizeof h);
    for(int i=1;i<=m;i++){
        cin>>a>>b>>c;
        add(a,b,c,i);
        add(b,a,c,i);
    }
    dj();
    memset(st, 0, sizeof st);
    dfs(1);
    cout<<ans.size()<<endl;
    for(auto x:ans)printf("%d ",x);
}