牛客算法周周练5

小雨坐地铁

题解：采用分层图跑最短路。

#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define ll long long
#define INF 0x3f3f3f3f
const int N = 1e6+100;
int n, m, s, t;
int ne[N<<1],head[N],e[N<<1],w[N<<1];
int vis[N],cnt;
ll dis[N];
void add(int a,int b,int c)
{
    e[cnt]=b,w[cnt]=c,ne[cnt]=head[a],head[a]=cnt++;
}
void Dijsktra(){
    memset(dis,INF,sizeof dis);
    memset(vis, 0 , sizeof vis );
    priority_queue< pair<int,int> ,vector< pair<int,int> > ,greater< pair<int,int> > > q;
    q.push(mp(0,m*n+s));
    dis[m*n+s] = 0;
    while(!q.empty()){
        int u, v, wi;
        u=q.top().second; q.pop();
        if(vis[u]) continue;
        vis[u] = 1;
        for(int i = head[u]; i != -1; i = ne[i]){
            v=e[i], wi=w[i];
            if(dis[v] > dis[u] + wi){
                dis[v] = dis[u] + wi;
                q.push(mp(dis[v],v));
            }
        }
    }
}
int main(){
    ios::sync_with_stdio(false);
    cin >> n >> m >> s >> t;
    memset(head,-1,sizeof(head));
    for(int i = 0; i < m; i++){
        int a, b ,c;
        cin >> a >> b >> c;
        int u, v;
        for(int j = 0; j < c ; j++){
            cin >> v;
            if(j){
                add(i*n+u,i*n+v,b);
                add(i*n+v,i*n+u,b);
            }
            add(i*n+v, m*n+v, 0);
            add(m*n+v, i*n+v, a);
            u = v;
        }
    }
    Dijsktra();
    ll res=dis[m*n+t];
    if(res >= INF) res = -1;
    cout << res << endl;
    return 0;
}

序列最小化

题解：
因为是排列，所以n个数字都不一样，所以我们 需要改变n-1个数字，每次改变k-1个数字，所以答案就是(n-1)/(k-1)向上取整。

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int n,k; cin>>n>>k;
    for(int i=1;i<=n;i++) {
        int tmp; cin>>tmp;
    }
    cout<<ceil((double)(n-1)/(k-1))<<endl;
}

简单瞎搞题

题解：
bitset的简单运用，如果我们能够组成第xi个数字那么在二进制中第xi*xi位上就是1，最后有多少个一就有多少种类。

#include<bits/stdc++.h>
using namespace std;
signed main()
{
    bitset<1000010> a,b;
    int n; cin>>n;
    a[0]=1;
    for(int i=1;i<=n;i++){
        int l,r; cin>>l>>r;
        b.reset();
        for(int j=l;j<=r;j++) b|=a<<(j*j);
        a=b;
    }
    printf("%d\n",a.count());
}