题意:
有一个圈,上面有n个点每个点都有一个值,走到这个点就可以获得这个点的值,这个人啊第一步一定落在第一个点上,然后他按照顺序走,走到n就再走到1,这样转圈圈,然后m次询问,每次询问一个x,让你输出最少走多少步就可以得到x这个值,永远都不可能则输出-1.
分析:
得到序列的前缀和si,总和sum,很容易想到x = sum * k + si
,然后。。。官方题解就讲的很清楚
(同余的地方很重要)
然后,我这边借鉴了别人的题解,采用了map套一个pair<ll,int>
的vector的方法实现的
具体代码实现细节在代码注释中
AC代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll maxn = 1e5 + 5;
ll n, m, sum, x;
ll num[maxn];
pair<ll, int> tmp;
map<ll, int> vis;//在sum为0的情况下使用
map<ll, vector<pair<ll, int>>> mp;//mp[i]表示前缀模sum后为i的集合,是一个vector,所以说是集合,vector里存的是pair
//pair(j,k)表示前缀为j最早出现的下标,而对于mp[i]里的所有pair(j,k)都满足j模sum后为i
bool flag;//标记sum的状态
ll xx, id, ans;
int pos;
int main()
{
int t;
scanf("%d", &t);
while(t--)
{
scanf("%lld%lld", &n, &m);
flag = false;
for(int i = 1; i <= n; ++i)
{
scanf("%lld", &num[i]);
num[i] += num[i - 1];
}
sum = num[n];
if(sum < 0)//总和为负,直接全部取反
{
sum = -sum;
flag = true;
for(int i = 1; i <= n; ++i) num[i] = -num[i];
}
else if(sum == 0)//总和为0,判断起来比较简单,略
{
vis.clear();
for(int i = 1; i <= n; ++i)
if(vis.find(num[i]) == vis.end())
vis[num[i]] = i;
while(m--)
{
scanf("%lld", &x);
if(x == 0)
{
printf("0\n");
continue;
}
if(vis.find(x) == vis.end()) printf("-1\n");
else printf("%d\n", vis[x]);
}
continue;
}
mp.clear();
for(int i = 1; i <= n; ++i)
{
id = (num[i] % sum + sum) % sum;//注意这个取模方式,负数取模
mp[id].push_back({-num[i], i});//注意看,这push了一个pair,然后push的是-num[i],很巧妙
//我们需要找的是最后一个小于等于x的数,这样的话到时候我们lower_bound(-x)就可以
}
for(auto it = mp.begin(); it != mp.end(); ++it)
sort(it -> second.begin(), it -> second.end());//对每一个vector进行排序,会按照pair的first为第一关键字,second为第二关键排序的
while(m--)
{
scanf("%lld", &x);
if(flag) x = -x;
if(!x)
{
printf("0\n");
continue;
}
xx = (x % sum + sum) % sum;//查询的x的余数
if(mp.find(xx) == mp.end())//判断是否有一个前缀与x同余
{
printf("-1\n");
continue;
}
tmp = {-x, 0};
pos = lower_bound(mp[xx].begin(), mp[xx].end(), tmp) - mp[xx].begin();//一个vector的二分查找,上面写过push -num[i]的时候
if(pos == mp[xx].size()) printf("-1\n");//妹找着
else//找着了,根据x = si + sum * k很容易推出答案来
{
ans = mp[xx][pos].second;
ans += (x + mp[xx][pos].first) / sum * n;
printf("%lld\n", ans);
}
}
}
return 0;
}