题目
题意: 给定一个长度为n的数,v(l,r)表示这个数从(l,r)产生的值。对于m组询问,每组询问,求最小化l1后最小化l2,使得v(l1,l1+w-1)v(l,r)+v(l2,l2+w-1) = k (mod 9).
思路: 字符串哈希嘛。发现base数组是10再模9,相当于没乘。然后把所有长度为l的字符串哈希值按值存起来,也就9个数,之后9*9暴力枚举最小的l1、l2组合即可。
O(n+81 * m)
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N = 2e5+10;
const int mod = 9;
int n,m,k,w,T,p;
// int base[N];
int s[N];
vector<int> va[10];
int get(int l,int r)
{
int ans = (s[r] - s[l-1])%mod;
ans = (ans + mod)%mod;
return ans;
}
struct node{
int l,r;
bool operator<(const node&rhs)const
{
if(l!=rhs.l) return l < rhs.l;
return r < rhs.r;
}
};
void solve()
{
string ss; cin>>ss;
// cout<<ss<<"????\n";
// cin.tie();
n = ss.size();
for(int i=0;i<n;++i)
{
s[i+1] = s[i] * 10 + ss[i]-'0';
s[i+1] %= mod;
}
cin>>w>>m;
// cout<<w<<" "<<m<<"??\n";
for(int i=0;i<mod;++i) va[i].clear();
for(int i=1;i+w-1<=n;++i)
{
int x = get(i,i+w-1);
// cout<<x<<"??\n";
va[x].push_back(i);
}
for(int i=0;i<mod;++i) sort(va[i].begin(),va[i].end());
while(m--)
{
int l,r,k; cin>>l>>r>>k;
int wh = get(l,r);
node ans = {1000000,1000000};
for(int i=0;i<mod;++i)
{
for(int j=0;j<mod;++j)
{
if((wh*i+j)%mod==k)
{
if(i==j)
{
if(va[i].size()<=1) continue;
node t = {va[i][0],va[i][1]};
if(t < ans) ans = t;
}
else
{
if(!va[i].size()||!va[j].size()) continue;
node t = {va[i][0],va[j][0]};
if(t < ans) ans = t;
}
}
}
}
if(ans.l==1000000) ans.l = ans.r = -1;
cout<<ans.l<<" "<<ans.r<<"\n";
}
}
signed main(void)
{
// base[0] = 1;
ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
cin>>T;
while(T--) solve();
return 0;
}