给定一个整数 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);
}