1005 Monopoly 同余求解（2021中国大学生程序设计竞赛CCPC-网络选拔赛重赛）

传送门

 题意： 告诉你一个长度为n的环及每个点的权值，从起点0出发，经过一个点时精力就加上其权值；现问达到精力x需要最少走多少步。

思路：

        · 后面才发现自己赛中的思路是错的：很明显 x = y + k*sum (y就是一个对应的(pre[j]%sum+sum)%sum，而显然 ans = k*n+j)，我是先利用 x/sum求到对应的 k，再判断是否有相应的 pre[j]存在；但是显然这种方法存在错误，因为不管怎样 j<=n 是必然的，所有我们这样求到的 k 肯定是最大k，只有先让 k 最小才能得到 min答案。

        · 所有正确的思路应该是按照所有y为第一维，去存所有可能的pre[j]；当输入x时找到对应的 y = (x%sum+sum)%sum，而其中最大的那个pre[j]便能得到最小的答案。

        · 参考博客，官方题解如下：

 代码实现：

//#define LOCAL

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x3f3f3f3f3f3f3f3f;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9+7;
const int  N = 2e5 + 5;

inline void read(int &x){
    char t=getchar();
    while(!isdigit(t)) t=getchar();
    for(x=t^48,t=getchar();isdigit(t);t=getchar()) x=x*10+(t^48);
}

int T = 1;

int n, m, x, sum, ans, flag, pre[N];
map<int, int> pos;
map<int, vector<pii>> mp;

signed main()
{
    IOS;

    #ifdef LOCAL
    freopen("in.txt", "r", stdin);
    #endif  //LOCAL

    cin >> T;

    while(T --){
        cin >> n >> m;
        for(int i = 1; i <= n; i ++){
            cin >> pre[i];
            pre[i] += pre[i-1];
        }
        sum = pre[n], flag = 0;
        if(!sum){            // 当sum为0的时候,直接寻找满足条件的pre[i]即可
            pos.clear();
            for(int i = 1; i <= n; i ++){
                if(!pos[pre[i]]) pos[pre[i]] = i;
            }
            while(m --){
                cin >> x;
                if(!x){ cout << 0 << endl; continue;}
                cout << (pos[x]?pos[x]:-1) << endl;
            }
            continue;
        }
        else if(sum<0){     // 当sum<0时全部取反计算
            sum = -sum;
            flag = 1;
            for(int i = 1; i <= n; i ++) pre[i] = -pre[i];
        }
        mp.clear();
        for(int i = 1; i <= n; i ++){
            int y = (pre[i]%sum+sum)%sum;
            mp[y].push_back({-pre[i], i});  //统计所有y对应可行的pre[j]
        }
        for(auto it = mp.begin(); it != mp.end(); it ++){
            sort(it->second.begin(), it->second.end()); //将对应pre[j]数组排序，方便利用二分函数直接查找
        }
        while(m --){
            cin >> x;
            if(!x){ cout << 0 << endl; continue;}
            if(flag) x = -x;
            int y = (x%sum+sum)%sum;
            if(mp.find(y)==mp.end()){ cout << -1 << endl; continue;} //如果对应的y不存在，说明此情况无解
            pii tmp = {-x, 0};
            int poss = lower_bound(mp[y].begin(), mp[y].end(), tmp)-mp[y].begin(); //寻找最大的pre[j]
            if(poss==mp[y].size()){ cout << -1 << endl; continue;}
            ans = mp[y][poss].second;
            ans += (x+mp[y][poss].first)/sum*n;
            cout << ans << endl;
        }
    }

    return 0;
}